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