def test6(self): instance = ph4Interface() # (debugger="xterm") instance.initialize_code() instance.set_eps2(0.0 ** 2) instance.set_eta(0.01) instance.commit_parameters() instance.new_particle( [1.0, 1.0, 1.0], [1.0, 0.0, -1.0], [0.0, 0.0, 0.0], [0.0, 0.0, 0.0], [0.0, 1.0, 0.0], [0.0, 0.0, 0.0], [0.0, 0.0, 0.0], ) instance.commit_particles() Ep = instance.get_potential_energy()["potential_energy"] Ek = instance.get_kinetic_energy()["kinetic_energy"] self.assertEqual(Ek, 0.5) self.assertEqual(Ep, -2.5) instance.delete_particle(2) instance.recommit_particles() n = instance.get_number_of_particles()["number_of_particles"] Ep = instance.get_potential_energy()["potential_energy"] Ek = instance.get_kinetic_energy()["kinetic_energy"] instance.cleanup_code() instance.stop() self.assertEqual(n, 2) self.assertEqual(Ek, 0.0) self.assertEqual(Ep, -0.5)
def test6(self): instance = ph4Interface()#(debugger="xterm") instance.initialize_code() instance.set_eps2(0.0**2) instance.set_eta(0.01) instance.commit_parameters() instance.new_particle( [1.0,1.0,1.0], [1.0,0.0,-1.0], [0.0,0.0,0.0], [0.0,0.0,0.0], [0.0,1.0,0.0], [0.0,0.0,0.0], [0.0,0.0,0.0] ) instance.commit_particles() Ep=instance.get_potential_energy()['potential_energy'] Ek=instance.get_kinetic_energy()['kinetic_energy'] self.assertEqual( Ek, 0.5) self.assertEqual( Ep, -2.5) instance.delete_particle(2) instance.recommit_particles() n=instance.get_number_of_particles()['number_of_particles'] Ep=instance.get_potential_energy()['potential_energy'] Ek=instance.get_kinetic_energy()['kinetic_energy'] instance.cleanup_code() instance.stop() self.assertEqual( n, 2) self.assertEqual( Ek, 0.) self.assertEqual( Ep, -0.5)
def test2(self): instance = ph4Interface() instance.initialize_code() for x in [0.101, 4.0]: error = instance.set_eps2(x) self.assertEquals(error, 0) value, error = instance.get_eps2() self.assertEquals(error, 0) self.assertEquals(x, value) instance.cleanup_code() instance.stop()
def test5(self): instance = ph4Interface() instance.initialize_code() instance.set_eta(0.01) n = 4000 ids = [i for i in range(1, n)] values = [1.0 * i for i in range(1, n)] instance.new_particle(values, values, values, values, values, values, values, values) error = instance.commit_particles() retrieved_state = instance.get_state(1) self.assertEquals(1.0, retrieved_state["mass"]) retrieved_state = instance.get_state(3999) instance.cleanup_code() instance.stop() self.assertEquals(3999.0, retrieved_state["mass"])
def test1(self): instance = ph4Interface() instance.initialize_code() instance.set_eta(0.01) index, error = instance.new_particle(11.0, 0.0, 0.0, 0.0, 1.0, 1.0, 1.0, 2.0) self.assertEquals(error, 0) self.assertEquals(index, 1) error = instance.commit_particles() self.assertEquals(error, 0) retrieved_state = instance.get_state(index) self.assertEquals(retrieved_state["__result"], 0) self.assertEquals(11.0, retrieved_state["mass"]) self.assertEquals(2.0, retrieved_state["radius"]) self.assertEquals(instance.get_number_of_particles()["number_of_particles"], 1) instance.cleanup_code() instance.stop()
def test1(self): instance = ph4Interface() instance.initialize_code() instance.set_eta(0.01) index, error = instance.new_particle(11.0, 0.0, 0.0, 0.0, 1.0, 1.0, 1.0, 2.0) self.assertEquals(error, 0) self.assertEquals(index, 1) error = instance.commit_particles() self.assertEquals(error, 0) retrieved_state = instance.get_state(index) self.assertEquals(retrieved_state['__result'], 0) self.assertEquals(11.0, retrieved_state['mass']) self.assertEquals(2.0, retrieved_state['radius']) self.assertEquals(instance.get_number_of_particles()['number_of_particles'], 1) instance.cleanup_code() instance.stop()
def test7(self): instance = ph4Interface() instance.initialize_code() instance.set_eps2(0.1 ** 2) instance.commit_parameters() id1, errorcode = instance.new_particle(mass=10.0, radius=1.0, x=0.0, y=0.0, z=0.0, vx=0.0, vy=0.0, vz=0.0) id2, errorcode = instance.new_particle(mass=10.0, radius=1.0, x=2.0, y=0.0, z=0.0, vx=10.0, vy=0.0, vz=0.0) instance.commit_particles() potential, errorcode = instance.get_potential(id1) self.assertEquals(errorcode, 0) self.assertAlmostRelativeEquals(potential, -10.0 / numpy.sqrt(2.0 ** 2 + 0.1 ** 2), 8) total_potential, errorcode = instance.get_potential_energy() potentials, errorcode = instance.get_potential([id1, id2]) instance.cleanup_code() instance.stop() self.assertAlmostRelativeEquals(total_potential, numpy.sum(potentials * [10.0, 10.0]) / 2.0)
def test7(self): instance = ph4Interface() instance.initialize_code() instance.set_eps2(0.1**2) instance.commit_parameters() id1,errorcode = instance.new_particle(mass = 10.0, radius = 1.0, x = 0.0, y = 0.0, z = 0.0, vx = 0.0, vy = 0.0, vz = 0.0) id2,errorcode = instance.new_particle(mass = 10.0, radius = 1.0, x = 2.0, y = 0.0, z = 0.0, vx = 10.0, vy = 0.0, vz = 0.0) instance.commit_particles() potential, errorcode = instance.get_potential(id1) self.assertEqual(errorcode, 0) self.assertAlmostRelativeEquals(potential, -10.0 / numpy.sqrt(2.0**2 + 0.1**2), 8) total_potential, errorcode = instance.get_potential_energy() potentials, errorcode = instance.get_potential([id1, id2]) instance.cleanup_code() instance.stop() self.assertAlmostRelativeEquals(total_potential, numpy.sum(potentials * [10.0, 10.0]) / 2.0)
def test3(self): instance = ph4Interface() instance.initialize_code() instance.set_eta(0.01) instance.new_particle([11.0,12.0,13.0,14.0] , [2.1,3.1,4.1,5.1] , [2.2,3.2,4.2,5.2] , [2.3,3.3,4.3,5.3] , [2.4,3.4,4.4,5.4] , [2.5,3.5,4.5,5.5] , [2.6,3.6,4.6,5.6] , [2.0,3.0,4.0,5.0]) error = instance.commit_particles() retrieved_state = instance.get_state(1) self.assertEquals(11.0, retrieved_state['mass']) retrieved_state = instance.get_state([2,3,4]) self.assertEquals(12.0, retrieved_state['mass'][0]) self.assertEquals(instance.get_number_of_particles()['number_of_particles'], 4) instance.cleanup_code() instance.stop()
def test9(self): instance = ph4Interface() instance.initialize_code() instance.set_eta(0.01) index, error = instance.new_particle(11.0, 0.0, 0.0, 0.0, 1.0, 1.0, 1.0, 2.0, 10) self.assertEquals(error, 0) self.assertEquals(index, 10) #index, error = instance.new_particle(12.0, 3.0, 0.0, 0.0, 0.0, 1.0, 1.0, 1.0, 10) #print index, error # self.assertEquals(error, -1) #self.assertEquals(index, 10) error = instance.commit_particles() self.assertEquals(error, 0) retrieved_state = instance.get_state(index) self.assertEquals(retrieved_state['__result'], 0) self.assertEquals(11.0, retrieved_state['mass']) self.assertEquals(2.0, retrieved_state['radius']) self.assertEquals(instance.get_number_of_particles()['number_of_particles'], 1) instance.cleanup_code() instance.stop()
def test5(self): instance = ph4Interface() instance.initialize_code() instance.set_eta(0.01) n = 4000 ids = [i for i in range(1,n)] values = [1.0 * i for i in range(1,n)] instance.new_particle( values , values , values , values , values , values , values , values) error = instance.commit_particles() retrieved_state = instance.get_state(1) self.assertEquals(1.0, retrieved_state['mass']) retrieved_state = instance.get_state(3999) instance.cleanup_code() instance.stop() self.assertEquals(3999.0, retrieved_state['mass'])
def test0(self): instance = ph4Interface() instance.initialize_code() instance.cleanup_code() instance.stop()