test_lambda_depraz.py
b87f34a0ce7emaster

test_lambda_depraz.py
View Options

			import h5py
			from numpy import *
			import PyCool.compute_lambda_interface as cl

			import Ptools as pt
			import sys

			print('launching python tests for cooling')

			#function asking for inputs

			#function arguments
			z = 2.0
			rho_H = 0.2
			nHe = 0.16
			metalicity = 0.001;

			cl.updateCoolingTable();
			# create vector of data
			T_min = 7e4
			T_max = 1e8
			dT = 1e3
			#T_interval = arange(T_min, T_max, dT)

			T_interval = 10**arange(log10(T_min),log10(T_max),0.1)


			cooling = zeros(len(T_interval),float32)
			for i,T in enumerate(T_interval):
			cooling[i] = fabs(cl.computeLambda(rho_H, T, nHe, metalicity))
			print " ",i,T,cooling[i]


			# plot data

			pt.plot(T_interval, cooling)
			pt.loglog()


			#pt.xlim([T_min, T_max])


			#pt.xlabel('T [K]')
			#pt.legend(('rho_H='+str(rho_H)+'\nz='+str(z),),loc='best')
			#pt.title('Normalized, net cooling rate ($\Lambda/n_H^2$ [erg s^-1 cm^3])\n')
			pt.show()

			#free allocated memory
			cl.freeMemory();

			print('program terminated')