def test1(self):
print("Test 1: initialization")
instance = SimpleX(**default_options)
instance.initialize_code()
instance.commit_parameters()
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()
def test1(self):
print "Test 1: initialization"
instance = SimpleX(**default_options)
instance.initialize_code()
instance.commit_parameters()
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 iliev_test_5( N=10000,
Ns=10,
L=15. | units.kpc,
dt=None):
"""
prepare iliev test and return SPH and simplex interfaces
"""
gas,sources=iliev_test_5_ic(N,Ns,L)
conv=nbody_system.nbody_to_si(1.0e9 | units.MSun, 1.0 | units.kpc)
sph=Fi(conv,use_gl=False,mode='periodic',redirection='none')
sph.initialize_code()
sph.parameters.use_hydro_flag=True
sph.parameters.radiation_flag=False
sph.parameters.self_gravity_flag=False
sph.parameters.gamma=1
sph.parameters.isothermal_flag=True
sph.parameters.integrate_entropy_flag=False
sph.parameters.timestep=dt
sph.parameters.verbosity=0
sph.parameters.pboxsize=2*L
sph.commit_parameters()
sph.gas_particles.add_particles(gas)
sph.commit_particles()
# sph.start_viewer()
rad=SimpleX(number_of_workers = 1,redirection='none')
rad.initialize_code()
rad.parameters.box_size=2*L
rad.parameters.hilbert_order=0
rad.commit_parameters()
gas.add_particles(sources)
rad.particles.add_particles(gas)
rad.commit_particles()
return sph,rad
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 iliev_test_5(N=10000, Ns=10, L=15. | units.kpc, dt=None):
"""
prepare iliev test and return SPH and simplex interfaces
"""
gas, sources = iliev_test_5_ic(N, Ns, L)
conv = nbody_system.nbody_to_si(1.0e9 | units.MSun, 1.0 | units.kpc)
sph = Fi(conv, use_gl=False, mode='periodic', redirection='none')
sph.initialize_code()
sph.parameters.use_hydro_flag = True
sph.parameters.radiation_flag = False
sph.parameters.self_gravity_flag = False
sph.parameters.gamma = 1
sph.parameters.isothermal_flag = True
sph.parameters.integrate_entropy_flag = False
sph.parameters.timestep = dt
sph.parameters.verbosity = 0
sph.parameters.pboxsize = 2 * L
sph.commit_parameters()
sph.gas_particles.add_particles(gas)
sph.commit_particles()
# sph.start_viewer()
rad = SimpleX(number_of_workers=1, redirection='none')
rad.initialize_code()
rad.parameters.box_size = 2 * L
rad.parameters.hilbert_order = 0
rad.commit_parameters()
gas.add_particles(sources)
rad.particles.add_particles(gas)
rad.commit_particles()
return sph, rad
def test4(self):
print("Test 4: default parameters")
instance = SimpleX(**default_options)
default = dict(timestep=0.05 | units.Myr,
source_effective_T=1.e5 | units.K,
hilbert_order=1,
number_of_freq_bins=1,
thermal_evolution_flag=0,
blackbody_spectrum_flag=0,
box_size=13200 | units.parsec,
metal_cooling_flag=0,
collisional_ionization_flag=0)
for x in default:
self.assertEqual(getattr(instance.parameters, x), default[x])
instance.commit_parameters()
for x in default:
self.assertEqual(getattr(instance.parameters, x), default[x])
tnow = instance.model_time
self.assertEqual(tnow, 0. | units.Myr)
instance.model_time = 321. | units.Myr
tnow = instance.model_time
self.assertEqual(tnow, 321. | units.Myr)
def test4(self):
print "Test 4: default parameters"
instance = SimpleX(**default_options)
default=dict( timestep= 0.05| units.Myr,
source_effective_T= 1.e5 | units.K,
hilbert_order= 1,
number_of_freq_bins= 1,
thermal_evolution_flag = 0,
blackbody_spectrum_flag = 0,
box_size=13200 | units.parsec,
metal_cooling_flag=0,
collisional_ionization_flag=0)
for x in default:
self.assertEqual(getattr(instance.parameters,x), default[x])
instance.commit_parameters()
for x in default:
self.assertEqual(getattr(instance.parameters,x), default[x])
tnow=instance.model_time
self.assertEqual(tnow, 0. | units.Myr)
instance.model_time=321. | units.Myr
tnow=instance.model_time
self.assertEqual(tnow, 321. | units.Myr)
