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()
