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 test1(self): print("Test 1: initialization") instance = SimpleXSplitSet(**default_options) instance.initialize_code() instance.commit_parameters() 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()
def test1(self): print "Test 1: initialization" instance = SimpleXSplitSet(**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 = 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.gas_particles.add_particles(particles) instance.src_particles.add_particles(src_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()