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.assertEquals(10.0, retrieved_state['mass'])
self.assertEquals(1, retrieved_state['radius'])
retrieved_state = hermite.get_state([0,1])
self.assertEquals(20.0, retrieved_state['mass'][1])
self.assertEquals(hermite.get_number_of_particles()['number_of_particles'], 2)
hermite.cleanup_code()
hermite.stop()
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.assertEquals(10.0, retrieved_state['mass'])
self.assertEquals(1, retrieved_state['radius'])
retrieved_state = hermite.get_state([0,1])
self.assertEquals(20.0, retrieved_state['mass'][1])
self.assertEquals(hermite.get_number_of_particles()['number_of_particles'], 2)
hermite.cleanup_code()
hermite.stop()
def test2(self):
instance = HermiteInterface()
instance.initialize_code()
for i in [0, 1, 2]:
temp_particle = instance.new_particle(mass = i, radius = 1.0, x = 0.0, y = 0.0, z = 0.0, vx = 0.0, vy = 0.0, vz = 0.0)
self.assertEquals(i, temp_particle['index_of_the_particle'])
instance.delete_particle(1)
self.assertEquals(2, instance.get_number_of_particles()['number_of_particles'])
self.assertEquals(0, instance.get_index_of_first_particle()['index_of_the_particle'])
self.assertEquals(2, instance.get_index_of_next_particle(0)['index_of_the_next_particle'])
self.assertEquals(0, instance.get_index_of_next_particle(0)['__result'])
self.assertEquals(-1, instance.get_index_of_next_particle(1)['__result'])
self.assertEquals(1, instance.get_index_of_next_particle(2)['__result'])
instance.stop()
def test2(self):
instance = HermiteInterface()
instance.initialize_code()
for i in [0, 1, 2]:
temp_particle = instance.new_particle(mass = i, radius = 1.0, x = 0.0, y = 0.0, z = 0.0, vx = 0.0, vy = 0.0, vz = 0.0)
self.assertEquals(i, temp_particle['index_of_the_particle'])
instance.delete_particle(1)
self.assertEquals(2, instance.get_number_of_particles()['number_of_particles'])
self.assertEquals(0, instance.get_index_of_first_particle()['index_of_the_particle'])
self.assertEquals(2, instance.get_index_of_next_particle(0)['index_of_the_next_particle'])
self.assertEquals(0, instance.get_index_of_next_particle(0)['__result'])
self.assertEquals(-1, instance.get_index_of_next_particle(1)['__result'])
self.assertEquals(1, instance.get_index_of_next_particle(2)['__result'])
instance.cleanup_code()
instance.stop()
