def test3(self):
print("Test 3: evolve")
instance = SimpleX(**default_options)
instance.initialize_code()
instance.commit_parameters()
input_file = os.path.join(os.path.dirname(__file__),
"test_simplex_data.txt")
particles = particles_from_input_file(input_file)
particles.du_dt = particles.u / (10 | units.Myr)
instance.particles.add_particles(particles)
# instance.particles.du_dt=particles.du_dt
# instance.commit_particles()
instance.particles.du_dt = particles.du_dt
self.assertAlmostEqual(instance.particles.xion.mean(), 0.0)
self.assertAlmostEqual(
instance.particles.du_dt.mean().in_(units.cm**2 / units.s**3),
particles.du_dt.mean().in_(units.cm**2 / units.s**3))
instance.evolve_model(0.5 | units.Myr)
self.assertAlmostEqual(
instance.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.particles.xion.mean(),
0.000845247683257)
instance.cleanup_code()
instance.stop()
def test1(self):
print "Test 1: initialization"
instance = SimpleX(**default_options)
instance.initialize_code()
instance.commit_parameters()
instance.cleanup_code()
instance.stop()
def test1(self):
print("Test 1: initialization")
instance = SimpleX(**default_options)
instance.initialize_code()
instance.commit_parameters()
instance.cleanup_code()
instance.stop()
def test8(self):
print "Test 8: two step evolve"
instance = SimpleX(**default_options)
input_file = os.path.join(os.path.dirname(__file__), "test_simplex_data.txt")
particles = particles_from_input_file(input_file)
instance.particles.add_particles(particles)
self.assertAlmostEqual(instance.particles.xion.mean(), 0.0)
instance.evolve_model(0.25 | units.Myr)
instance.evolve_model(0.5 | units.Myr)
self.assertEqual( instance.particles.flux.max().value_in(1.e48* units.s**-1), 5)
self.assertAlmostRelativeEqual(instance.particles.xion.mean(),0.00084660917243,3)
instance.cleanup_code()
instance.stop()
def test8(self):
print "Test 8: two step evolve"
instance = SimpleX(**default_options)
input_file = os.path.join(os.path.dirname(__file__), "test_simplex_data.txt")
particles = particles_from_input_file(input_file)
instance.particles.add_particles(particles)
self.assertAlmostEqual(instance.particles.xion.mean(), 0.0)
instance.evolve_model(0.25 | units.Myr)
instance.evolve_model(0.5 | units.Myr)
self.assertEqual( instance.particles.flux.max().value_in(1.e48* units.s**-1), 5)
self.assertAlmostRelativeEqual(instance.particles.xion.mean(),0.00084660917243,3)
instance.cleanup_code()
instance.stop()
def test2(self):
print "Test 2: commit_particles, getters and setters"
instance = SimpleX(**default_options)
instance.initialize_code()
instance.commit_parameters()
input_file = os.path.join(os.path.dirname(__file__), "test_simplex_data.txt")
particles = particles_from_input_file(input_file)
instance.particles.add_particles(particles)
instance.commit_particles()
# for attribute in ['position', 'rho', 'flux', 'xion']:
# self.assertAlmostEqual(13200.*getattr(particles, attribute),
# 13200.*getattr(instance.particles, attribute), 5)
# setattr(instance.particles, attribute, getattr(particles, attribute)/2.0)
# self.assertAlmostEqual(13200.*getattr(particles, attribute)/2.0,
# 13200.*getattr(instance.particles, attribute), 5)
instance.cleanup_code()
instance.stop()
def test2(self):
print "Test 2: commit_particles, getters and setters"
instance = SimpleX(**default_options)
instance.initialize_code()
instance.commit_parameters()
input_file = os.path.join(os.path.dirname(__file__), "test_simplex_data.txt")
particles = particles_from_input_file(input_file)
instance.particles.add_particles(particles)
instance.commit_particles()
# for attribute in ['position', 'rho', 'flux', 'xion']:
# self.assertAlmostEqual(13200.*getattr(particles, attribute),
# 13200.*getattr(instance.particles, attribute), 5)
# setattr(instance.particles, attribute, getattr(particles, attribute)/2.0)
# self.assertAlmostEqual(13200.*getattr(particles, attribute)/2.0,
# 13200.*getattr(instance.particles, attribute), 5)
instance.cleanup_code()
instance.stop()
def test7(self):
print "Test 7: two step evolve"
instance = SimpleX(**default_options)
instance.parameters.recombination_radiation_flag=1
input_file = os.path.join(os.path.dirname(__file__), "test_simplex_data.txt")
particles = particles_from_input_file(input_file)
instance.particles.add_particles(particles)
self.assertAlmostEqual(instance.particles.xion.mean(), 0.0)
instance.evolve_model(0.25 | units.Myr)
flux_before=instance.particles.flux.sum()
instance.particles.flux=particles.flux
flux_after=instance.particles.flux.sum()
self.assertEqual(flux_before,flux_after)
instance.evolve_model(0.5 | units.Myr)
self.assertAlmostRelativeEqual(instance.particles.xion.mean(),0.00084660917243,3)
self.assertEqual( instance.particles.flux.max().value_in(1.e48* units.s**-1), 5)
instance.cleanup_code()
instance.stop()
def test7(self):
print "Test 7: two step evolve"
instance = SimpleX(**default_options)
instance.parameters.recombination_radiation_flag=1
input_file = os.path.join(os.path.dirname(__file__), "test_simplex_data.txt")
particles = particles_from_input_file(input_file)
instance.particles.add_particles(particles)
self.assertAlmostEqual(instance.particles.xion.mean(), 0.0)
instance.evolve_model(0.25 | units.Myr)
flux_before=instance.particles.flux.sum()
instance.particles.flux=particles.flux
flux_after=instance.particles.flux.sum()
self.assertEqual(flux_before,flux_after)
instance.evolve_model(0.5 | units.Myr)
self.assertAlmostRelativeEqual(instance.particles.xion.mean(),0.00084660917243,3)
self.assertEqual( instance.particles.flux.max().value_in(1.e48* units.s**-1), 5)
instance.cleanup_code()
instance.stop()
def test3(self):
print "Test 3: evolve"
instance = SimpleX(**default_options)
instance.initialize_code()
instance.commit_parameters()
input_file = os.path.join(os.path.dirname(__file__), "test_simplex_data.txt")
particles = particles_from_input_file(input_file)
particles.du_dt = particles.u/(10|units.Myr)
instance.particles.add_particles(particles)
# instance.particles.du_dt=particles.du_dt
# instance.commit_particles()
instance.particles.du_dt=particles.du_dt
self.assertAlmostEqual(instance.particles.xion.mean(), 0.0)
self.assertAlmostEqual(instance.particles.du_dt.mean().in_(units.cm**2/units.s**3),particles.du_dt.mean().in_(units.cm**2/units.s**3))
instance.evolve_model(0.5 | units.Myr)
self.assertAlmostEqual(instance.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.particles.xion.mean(), 0.000845247683257)
instance.cleanup_code()
instance.stop()
