def test2(self): print("test2: test parameters") instance = SPHRay() self.assertEqual(instance.get_name_of_current_state(), 'UNINITIALIZED') for par, val in [("isothermal_flag", False), ("hydrogen_case_A_flag", True), ("helium_case_A_flag", True)]: val1 = getattr(instance.parameters, par) self.assertEqual(val, val1) setattr(instance.parameters, par, False) val1 = getattr(instance.parameters, par) self.assertEqual(val1, False) setattr(instance.parameters, par, True) val1 = getattr(instance.parameters, par) self.assertEqual(val1, True) setattr(instance.parameters, par, False) val1 = getattr(instance.parameters, par) self.assertEqual(val1, False) for par, val in [("ionization_temperature_solver", 2), ("boundary_condition", 0)]: val1 = getattr(instance.parameters, par) self.assertEqual(val, val1) setattr(instance.parameters, par, 123) val1 = getattr(instance.parameters, par) self.assertEqual(val1, 123) for par, val in [("spectra_file", "./spectra/thermal1e5.cdf")]: val1 = getattr(instance.parameters, par) self.assertEqual(val, val1) setattr(instance.parameters, par, "somefile") val1 = getattr(instance.parameters, par) self.assertEqual(val1, "somefile") for par, val, tval in [ ("number_of_rays", 1022.69032205 | (units.Myr**-1), 10000 | units.Myr**-1), ("box_size", 13.2 | units.kpc, 10. | units.kpc), ("default_spectral_type", -1., 1.) ]: val1 = getattr(instance.parameters, par) self.assertAlmostRelativeEqual(val, val1, 6) setattr(instance.parameters, par, tval) val1 = getattr(instance.parameters, par) self.assertAlmostRelativeEqual(val1, tval, 6) self.assertEqual(instance.get_name_of_current_state(), 'INITIALIZED') instance.stop()
def test2(self): print "test2: test parameters" instance=SPHRay() self.assertEquals(instance.get_name_of_current_state(), 'UNINITIALIZED') for par,val in [("isothermal_flag", False), ("hydrogen_case_A_flag", True),("helium_case_A_flag", True)]: val1=getattr(instance.parameters,par) self.assertEqual(val,val1) setattr(instance.parameters, par, False) val1=getattr(instance.parameters,par) self.assertEqual(val1,False) setattr(instance.parameters, par, True) val1=getattr(instance.parameters,par) self.assertEqual(val1,True) setattr(instance.parameters, par, False) val1=getattr(instance.parameters,par) self.assertEqual(val1,False) for par,val in [ ("ionization_temperature_solver", 2),("boundary_condition", 0)]: val1=getattr(instance.parameters,par) self.assertEqual(val,val1) setattr(instance.parameters, par, 123) val1=getattr(instance.parameters,par) self.assertEqual(val1,123) for par,val in [ ("spectra_file", "./spectra/thermal1e5.cdf")]: val1=getattr(instance.parameters,par) self.assertEqual(val,val1) setattr(instance.parameters, par, "somefile") val1=getattr(instance.parameters,par) self.assertEqual(val1,"somefile") for par,val,tval in [ ("number_of_rays", 1022.69032205 | (units.Myr**-1) , 10000| units.Myr**-1), ("box_size",13.2 | units.kpc, 10. | units.kpc), ("default_spectral_type", -1.,1.)]: val1=getattr(instance.parameters,par) self.assertAlmostRelativeEqual(val,val1,6) setattr(instance.parameters, par, tval) val1=getattr(instance.parameters,par) self.assertAlmostRelativeEqual(val1,tval,6) self.assertEquals(instance.get_name_of_current_state(), 'INITIALIZED') instance.stop()
def test1(self): print "test1: adding particles" instance=SPHRay() gasparts=self.read_gas_file(os.path.join(os.path.dirname(__file__), "test_sphray_data_4K")) srcparts=self.read_src_file(os.path.join(os.path.dirname(__file__), "test_sphray_data_sources_001.1")) self.assertEqual(len(instance.gas_particles),0) self.assertEqual(len(instance.src_particles),0) instance.gas_particles.add_particles(gasparts) instance.src_particles.add_particles(srcparts) self.assertEqual(len(instance.gas_particles),len(gasparts)) self.assertEqual(len(instance.src_particles),len(srcparts)) self.assertEquals(instance.get_name_of_current_state(), 'EDIT') gaspart2=instance.gas_particles.copy() self.assertAlmostRelativeEquals(gasparts.position,gaspart2.position,6) self.assertAlmostRelativeEquals(gasparts.rho,gaspart2.rho,6) self.assertAlmostRelativeEquals(gasparts.u,gaspart2.u,6) instance.cleanup_code() instance.stop()
def test0(self): print "test1: basic startup and flow" instance=SPHRay() self.assertEquals(instance.get_name_of_current_state(), 'UNINITIALIZED') instance.initialize_code() self.assertEquals(instance.get_name_of_current_state(), 'INITIALIZED') instance.parameters.box_size = 100 | units.parsec self.assertAlmostRelativeEquals(instance.parameters.box_size, 100 | units.parsec,7) instance.commit_parameters() self.assertEquals(instance.get_name_of_current_state(), 'EDIT') instance.commit_particles() self.assertEquals(instance.get_name_of_current_state(), 'RUN') self.assertAlmostRelativeEquals(instance.parameters.box_size, 100 | units.parsec,7) instance.cleanup_code() instance.stop()