def test_calc_T(): from myscience import wolfire03 as w03 n = 10 # cm^-3 T = w03.calc_T(n) assert T < 188 and T > 168
def test_calc_T_asarray(): from myscience import wolfire03 as w03 n = (10,10) # cm^-3 T = w03.calc_T(n) assert all(T < 188) and all(T > 168)
def test_w03Temp_vs_pressureTemp(): import matplotlib.pyplot as plt from myscience import wolfire03 as w03 from myscience import calc_temperature import numpy as np # range of densities to compare n = np.logspace(0,2,100) # get Wolfire+03 temperatures for several rad fields G_0_list = [0.6, 0.8, 0.64,] cloud_list = ['California', 'Perseus', 'Taurus'] temp_w03_list = [] for G_0 in G_0_list: temp_w03_list.append(w03.calc_T(n, G_0=G_0)) # get temperatures assumes regular pressure equilibrium temp_pressure = calc_temperature(n_H=n, calc_error=False) # Plot the figure # ----------------------------------------------------------------------------- figure = plt.figure(figsize=(5,5)) for i in xrange(len(G_0_list)): G_0 = G_0_list[i] cloud = cloud_list[i] temp_w03 = temp_w03_list[i] plt.plot(n, temp_w03, linestyle='--', linewidth='2', label='W+03, $G_0$={0:.2f}, {1:s}'.format(G_0, cloud), ) plt.plot(n, temp_pressure, linestyle='-', linewidth=2, label='Ideal Gas Law', ) plt.xlabel(r'$n$ [cm$^{-3}$]') plt.ylabel(r'$T$ [K]') plt.xscale('log') plt.yscale('log') plt.legend(loc='best') plt.savefig('test_plots/myscience_test.test_w03Temp_vs_pressureTemp.png')