def test3(self):
huayno = HuaynoInterface()
huayno.eps2 = 0.101
self.assertEquals(0.101, huayno.eps2)
huayno.eps2 = 0.110
self.assertEquals(0.110, huayno.eps2)
huayno.stop()
def test2(self):
instance = HuaynoInterface()
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.assertEqual(i, temp_particle['index_of_the_particle'])
instance.delete_particle(1)
self.assertEqual(
2,
instance.get_number_of_particles()['number_of_particles'])
self.assertEqual(
0,
instance.get_index_of_first_particle()['index_of_the_particle'])
self.assertEqual(
2,
instance.get_index_of_next_particle(0)
['index_of_the_next_particle'])
self.assertEqual(0, instance.get_index_of_next_particle(0)['__result'])
self.assertEqual(-2,
instance.get_index_of_next_particle(1)['__result'])
self.assertEqual(1, instance.get_index_of_next_particle(2)['__result'])
instance.cleanup_code()
instance.stop()
def test1(self):
instance = HuaynoInterface()
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'])
instance.cleanup_code()
instance.stop()
def test3(self):
huayno = HuaynoInterface()
huayno.eps2 = 0.101
self.assertEquals(0.101, huayno.eps2)
huayno.eps2 = 0.110
self.assertEquals(0.110, huayno.eps2)
huayno.stop()
def test3(self):
huayno = HuaynoInterface()
huayno.set_eps2(0.101)
self.assertEqual(0.101, huayno.get_eps2()['eps2'])
huayno.set_eps2(0.110)
self.assertEqual(0.110, huayno.get_eps2()['eps2'])
huayno.cleanup_code()
huayno.stop()
def test5(self):
huayno = HuaynoInterface()
huayno.initialize_code()
huayno.new_particle([10,20],[0,0],[0,0], [0,0], [0,0], [0,0], [0,0],[1,1])
retrieved_state = huayno.get_state(0)
self.assertEqual(10.0, retrieved_state['mass'])
self.assertEqual(1, retrieved_state['radius'])
retrieved_state = huayno.get_state([0,1])
self.assertEqual(20.0, retrieved_state['mass'][1])
self.assertEqual(huayno.get_number_of_particles()['number_of_particles'], 2)
huayno.cleanup_code()
huayno.stop()
