def test5(self): print("Test 2: delete particles") instance = SimpleXInterface(**default_options) self.assertEqual( 0, instance.set_simplex_output_directory(instance.output_directory)) self.assertEqual(0, instance.initialize_code()) self.assertEqual(0, instance.commit_parameters()) input_file = os.path.join(os.path.dirname(__file__), "test_simplex_data.txt") x, y, z, n_H, flux, X_ion, u = read_input_file(input_file) x = numpy.array(x) y = numpy.array(y) z = numpy.array(z) number_of_particles = len(x) indices, errors = instance.new_particle(x, y, z, n_H, flux, X_ion, u) self.assertEqual(errors, [0] * number_of_particles) self.assertEqual(indices, list(range(number_of_particles))) error = instance.delete_particle(indices[0]) self.assertEqual(error, -1) instance.commit_particles() error = instance.delete_particle(indices[0]) # this one I don't understand: self.assertEqual(error, -1) self.assertEqual(0, instance.evolve_model(0.125)) error = instance.delete_particle(indices[0]) self.assertEqual(error, 0)
def test5(self): print "Test 2: delete particles" instance = SimpleXInterface(**default_options) self.assertEqual(0, instance.set_simplex_output_directory(instance.output_directory)) self.assertEqual(0, instance.initialize_code()) self.assertEqual(0, instance.commit_parameters()) input_file = os.path.join(os.path.dirname(__file__), "test_simplex_data.txt") x, y, z, n_H, flux, X_ion,u = read_input_file(input_file) x=numpy.array(x) y=numpy.array(y) z=numpy.array(z) number_of_particles = len(x) indices, errors = instance.new_particle(x, y, z, n_H, flux, X_ion,u) self.assertEqual(errors, [0]*number_of_particles) self.assertEqual(indices, range(number_of_particles)) error=instance.delete_particle(indices[0]) self.assertEqual(error, -1) instance.commit_particles() error=instance.delete_particle(indices[0]) # this one I don't understand: self.assertEqual(error, -1) self.assertEqual(0, instance.evolve_model(0.125)) error=instance.delete_particle(indices[0]) self.assertEqual(error, 0)
def test3(self): print("Test 3: evolve") instance = SimpleXInterface(**default_options) self.assertEqual( 0, instance.set_simplex_output_directory(instance.output_directory)) self.assertEqual(0, instance.initialize_code()) self.assertEqual(0, instance.commit_parameters()) input_file = os.path.join(os.path.dirname(__file__), "test_simplex_data.txt") x, y, z, n_H, flux, X_ion, u = read_input_file(input_file) number_of_particles = len(x) indices, errors = instance.new_particle(x, y, z, n_H, flux, X_ion, u) self.assertEqual(errors, [0] * number_of_particles) self.assertEqual(indices, list(range(number_of_particles))) self.assertEqual(0, instance.commit_particles()) X_ion, errors = instance.get_ionisation(indices) self.assertEqual(errors, [0] * number_of_particles) self.assertAlmostEqual(X_ion.sum() / number_of_particles, 0.0) density, errors = instance.get_density(indices) self.assertEqual(errors, [0] * number_of_particles) self.assertAlmostEqual(density.sum(), 1.0, 6) self.assertEqual(0, instance.evolve_model(0.5)) density, errors = instance.get_density(indices) self.assertEqual(errors, [0] * number_of_particles) self.assertAlmostEqual(density.sum(), 1.0, 6) flux, errors = instance.get_flux(indices) self.assertEqual(errors, [0] * number_of_particles) self.assertEqual(flux.sum(), 5.0) X_ion, errors = instance.get_ionisation(indices) self.assertEqual(errors, [0] * number_of_particles) self.assertAlmostEqual(X_ion.sum() / number_of_particles, 0.000845247683257) self.assertEqual(0, instance.cleanup_code()) instance.stop()
def test3(self): print "Test 3: evolve" instance = SimpleXInterface(**default_options) self.assertEqual(0, instance.set_simplex_output_directory(instance.output_directory)) self.assertEqual(0, instance.initialize_code()) self.assertEqual(0, instance.commit_parameters()) input_file = os.path.join(os.path.dirname(__file__), "test_simplex_data.txt") x, y, z, n_H, flux, X_ion,u = read_input_file(input_file) number_of_particles = len(x) indices, errors = instance.new_particle(x, y, z, n_H, flux, X_ion,u) self.assertEqual(errors, [0]*number_of_particles) self.assertEqual(indices, range(number_of_particles)) self.assertEqual(0, instance.commit_particles()) X_ion, errors = instance.get_ionisation(indices) self.assertEqual(errors, [0]*number_of_particles) self.assertAlmostEqual(X_ion.sum()/number_of_particles, 0.0) density, errors = instance.get_density(indices) self.assertEqual(errors, [0]*number_of_particles) self.assertAlmostEqual(density.sum(), 1.0,6) self.assertEqual(0, instance.evolve_model(0.5)) density, errors = instance.get_density(indices) self.assertEqual(errors, [0]*number_of_particles) self.assertAlmostEqual(density.sum(), 1.0,6) flux, errors = instance.get_flux(indices) self.assertEqual(errors, [0]*number_of_particles) self.assertEqual(flux.sum(), 5.0) X_ion, errors = instance.get_ionisation(indices) self.assertEqual(errors, [0]*number_of_particles) self.assertAlmostEqual(X_ion.sum()/number_of_particles, 0.000845247683257) self.assertEqual(0, instance.cleanup_code()) instance.stop()