def test3(self):
print("Test 3: evolve")
instance = SimpleXSplitSet(**default_options)
instance.initialize_code()
instance.commit_parameters()
input_file = os.path.join(os.path.dirname(__file__),
"test_simplex_data.txt")
particles, src_particles = splitset_from_input_file(input_file)
instance.src_particles.add_particles(src_particles)
particles.du_dt = particles.u / (10 | units.Myr)
instance.gas_particles.add_particles(particles)
self.assertAlmostEqual(instance.gas_particles.xion.mean(), 0.0)
self.assertAlmostEqual(
instance.gas_particles.du_dt.mean().in_(units.cm**2 / units.s**3),
particles.du_dt.mean().in_(units.cm**2 / units.s**3))
self.assertEqual(instance.get_name_of_current_state(), 'EDIT')
instance.evolve_model(0.5 | units.Myr)
self.assertEqual(instance.get_name_of_current_state(), 'RUN')
self.assertAlmostEqual(
instance.gas_particles.du_dt.mean().in_(units.cm**2 / units.s**3),
particles.du_dt.mean().in_(units.cm**2 / units.s**3))
self.assertAlmostEqual(instance.gas_particles.xion.mean(),
0.000845247683257)
instance.gas_particles.remove_particles(particles[0:4])
# this is what we would like....
# self.assertEquals(instance.get_name_of_current_state(), 'UPDATE')
# instance.recommit_particles()
instance.evolve_model(0.75 | units.Myr)
self.assertEqual(len(instance.particles), len(particles) - 4)
instance.cleanup_code()
instance.stop()
def test3(self):
print "Test 3: evolve"
instance = SimpleXSplitSet(**default_options)
instance.initialize_code()
instance.commit_parameters()
input_file = os.path.join(os.path.dirname(__file__), "test_simplex_data.txt")
particles,src_particles = splitset_from_input_file(input_file)
instance.src_particles.add_particles(src_particles)
particles.du_dt = particles.u/(10|units.Myr)
instance.gas_particles.add_particles(particles)
self.assertAlmostEqual(instance.gas_particles.xion.mean(), 0.0)
self.assertAlmostEqual(instance.gas_particles.du_dt.mean().in_(units.cm**2/units.s**3),particles.du_dt.mean().in_(units.cm**2/units.s**3))
self.assertEquals(instance.get_name_of_current_state(), 'EDIT')
instance.evolve_model(0.5 | units.Myr)
self.assertEquals(instance.get_name_of_current_state(), 'RUN')
self.assertAlmostEqual(instance.gas_particles.du_dt.mean().in_(units.cm**2/units.s**3),particles.du_dt.mean().in_(units.cm**2/units.s**3))
self.assertAlmostEqual(instance.gas_particles.xion.mean(), 0.000845247683257)
instance.gas_particles.remove_particles(particles[0:4])
# this is what we would like....
# self.assertEquals(instance.get_name_of_current_state(), 'UPDATE')
# instance.recommit_particles()
instance.evolve_model(0.75 | units.Myr)
self.assertEqual(len(instance.particles), len(particles)-4)
instance.cleanup_code()
instance.stop()
