def test5(self): hermite = HermiteInterface() hermite.initialize_code() hermite.new_particle([10, 20], [0, 0], [0, 0], [0, 0], [0, 0], [0, 0], [0, 0], [1, 1]) retrieved_state = hermite.get_state(0) self.assertEqual(10.0, retrieved_state['mass']) self.assertEqual(1, retrieved_state['radius']) retrieved_state = hermite.get_state([0, 1]) self.assertEqual(20.0, retrieved_state['mass'][1]) self.assertEqual( hermite.get_number_of_particles()['number_of_particles'], 2) hermite.cleanup_code() hermite.stop()
def test6(self): hermite = HermiteInterface() hermite.initialize_code() hermite.new_particle([10,10],[-1,1],[0,0], [0,0], [0,0], [0,0], [0,0], [1,1]) retrieved_state = hermite.get_state(0) retr = hermite.get_potential_at_point(0.01, 0,0,0) self.assertEqual(retr['phi'], -20.0) hermite.cleanup_code() hermite.stop()
def test1(self): instance = HermiteInterface() instance.initialize_code() res1 = instance.new_particle(mass=11.0, radius=2.0, x=0.0, y=0.0, z=0.0, vx=0.0, vy=0.0, vz=0.0) res2 = instance.new_particle(mass=21.0, radius=5.0, x=10.0, y=0.0, z=0.0, vx=10.0, vy=0.0, vz=0.0) self.assertEqual(0, res1['index_of_the_particle']) self.assertEqual(1, res2['index_of_the_particle']) retrieved_state1 = instance.get_state(0) retrieved_state2 = instance.get_state(1) self.assertEqual(11.0, retrieved_state1['mass']) self.assertEqual(21.0, retrieved_state2['mass']) self.assertEqual(0.0, retrieved_state1['x']) self.assertEqual(10.0, retrieved_state2['x']) self.assertEqual(2.0, retrieved_state1['radius']) self.assertEqual(5.0, retrieved_state2['radius']) instance.cleanup_code() instance.stop()