def test5(self):
smalln = SmallNInterface()
smalln.initialize_code()
smalln.new_particle([10,20],[0,0],[0,0], [0,0], [0,0], [0,0], [0,0],[1,1])
retrieved_state = smalln.get_state(1)
self.assertEquals(10.0, retrieved_state['mass'])
self.assertEquals(1, retrieved_state['radius'])
retrieved_state = smalln.get_state([1,2])
self.assertEquals(20.0, retrieved_state['mass'][1])
self.assertEquals(smalln.get_number_of_particles()['number_of_particles'], 2)
smalln.cleanup_code()
smalln.stop()
def test5(self):
smalln = SmallNInterface()
smalln.initialize_code()
smalln.new_particle([10, 20], [0, 0], [0, 0], [0, 0], [0, 0], [0, 0],
[0, 0], [1, 1])
retrieved_state = smalln.get_state(1)
self.assertEqual(10.0, retrieved_state['mass'])
self.assertEqual(1, retrieved_state['radius'])
retrieved_state = smalln.get_state([1, 2])
self.assertEqual(20.0, retrieved_state['mass'][1])
self.assertEqual(
smalln.get_number_of_particles()['number_of_particles'], 2)
smalln.cleanup_code()
smalln.stop()
def test6(self):
smalln = SmallNInterface()
smalln.initialize_code()
smalln.new_particle([10,10],[-1,1],[0,0], [0,0], [0,0], [0,0], [0,0], [1,1])
retrieved_state = smalln.get_state(1)
self.assertFalse(hasattr(smalln, 'get_potential_at_point'))
#retr = smalln.get_potential_at_point(0.01, 0, 0, 0)
#self.assertEqual(retr['__result'], -1)
smalln.cleanup_code()
smalln.stop()
def test6(self):
smalln = SmallNInterface()
smalln.initialize_code()
smalln.new_particle([10, 10], [-1, 1], [0, 0], [0, 0], [0, 0], [0, 0],
[0, 0], [1, 1])
retrieved_state = smalln.get_state(1)
self.assertFalse(hasattr(smalln, 'get_potential_at_point'))
#retr = smalln.get_potential_at_point(0.01, 0, 0, 0)
#self.assertEqual(retr['__result'], -1)
smalln.cleanup_code()
smalln.stop()
def test1(self):
instance = SmallNInterface()
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.assertEquals(1, res1['index_of_the_particle'])
self.assertEquals(2, res2['index_of_the_particle'])
retrieved_state1 = instance.get_state(1)
retrieved_state2 = instance.get_state(2)
self.assertEquals(11.0, retrieved_state1['mass'])
self.assertEquals(21.0, retrieved_state2['mass'])
self.assertEquals(0.0, retrieved_state1['x'])
self.assertEquals(10.0, retrieved_state2['x'])
self.assertEquals(2.0, retrieved_state1['radius'])
self.assertEquals(5.0, retrieved_state2['radius'])
instance.cleanup_code()
instance.stop()
def test1(self):
instance = SmallNInterface()
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(1, res1['index_of_the_particle'])
self.assertEqual(2, res2['index_of_the_particle'])
retrieved_state1 = instance.get_state(1)
retrieved_state2 = instance.get_state(2)
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()
