def test4(self): instance=SPHRay() instance.gas_particles.add_particle( Particle( mass = 1 | (10**10*units.MSun), x = 2 | units.kpc, y = 3 | units.kpc, z = 4 | units.kpc, h_smooth = 0.1 | (units.kpc), rho = 0.5 | ((10**10*units.MSun) /(units.kpc)**3), xion = 0.01, u = 0.2 | (10**5 * units.cm/units.s)**2 ) ) instance.src_particles.add_particle( Particle( luminosity = 1 | 1e48 * units.s**-1, x = 2 | units.m, y = 3 | units.m, z = 4 | units.m, SpcType = 12.3 ) ) instance.commit_particles() self.assertAlmostRelativeEquals(instance.src_particles.luminosity, 1 | 1e48 * units.s**-1, 6) self.assertAlmostRelativeEquals(instance.src_particles.x, 2 | units.m,7) self.assertAlmostRelativeEquals(instance.src_particles.y, 3 | units.m,7) self.assertAlmostRelativeEquals(instance.src_particles.z, 4 | units.m,7) self.assertAlmostRelativeEquals(instance.src_particles.SpcType, 12.3,7) print instance.src_particles 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()