Two star test



In [1]:

    
%matplotlib inline
import matplotlib.pyplot as plt
import numpy as np
from scipy.integrate import odeint
from IPython.html.widgets import interact, interactive, fixed









    



:0: FutureWarning: IPython widgets are experimental and may change in the future.



In [2]:

    
from plotting_function import plotter



In [3]:

    
from initial_velocities import velocities_m, velocities_S



In [4]:

    
from DE_solver import derivs, equationsolver

Defining some test values for a simple two star system to check if everything was working correctly:



In [5]:

    
max_time_test = 1
time_step_test = 80
M_test = 1e11
S_test = 1e11
S_y_test = 70
S_x_test = -.01*S_y_test**2+25
m_x_test_1 = -3.53
m_y_test_1 = 3.53
m_x_test_2 = -3.53
m_y_test_2 = -3.53



In [6]:

    
vxS_test = velocities_S(M_test,S_test,S_x_test,S_y_test)[0]
vyS_test = velocities_S(M_test,S_test,S_x_test,S_y_test)[1]
vxm_test_1 = velocities_m(M_test,m_x_test_1,m_y_test_1)[0]
vym_test_1 = velocities_m(M_test,m_x_test_1,m_y_test_1)[1]
vxm_test_2 = velocities_m(M_test,m_x_test_2,m_y_test_2)[0]
vym_test_2 = velocities_m(M_test,m_x_test_2,m_y_test_2)[1]



In [7]:

    
ic_test = np.array([S_x_test,S_y_test,vxS_test,vyS_test,m_x_test_1,m_y_test_1,vxm_test_1,vym_test_1,
                    m_x_test_2,m_y_test_2,vxm_test_2,vym_test_2])

Using equationsolver to solve the DE's



In [8]:

    
sol_test = equationsolver(ic_test,max_time_test,time_step_test,M_test,S_test)

Saving results and initial conditions to disk



In [9]:

    
np.savez('two_star_test_sol+ic.npz',sol_test,ic_test)



In [ ]: