def test5(self): interface = BHTreeInterface() interface.initialize_code() interface.commit_parameters() interface.new_particle([10,20],[0,0],[0,0], [0,0], [0,0], [0,0], [0,0],[1,1]) interface.commit_particles() retrieved_state = interface.get_state(1) self.assertEquals(10.0, retrieved_state['mass']) self.assertEquals(1, retrieved_state['radius']) retrieved_state = interface.get_state([1,2]) self.assertEquals(20.0, retrieved_state['mass'][1]) self.assertEquals(interface.get_number_of_particles()['number_of_particles'], 2) interface.cleanup_code() interface.stop()
def test5(self): interface = BHTreeInterface() interface.initialize_code() interface.commit_parameters() interface.new_particle([10, 20], [0, 0], [0, 0], [0, 0], [0, 0], [0, 0], [0, 0], [1, 1]) interface.commit_particles() retrieved_state = interface.get_state(1) self.assertEquals(10.0, retrieved_state['mass']) self.assertEquals(1, retrieved_state['radius']) retrieved_state = interface.get_state([1, 2]) self.assertEquals(20.0, retrieved_state['mass'][1]) self.assertEquals( interface.get_number_of_particles()['number_of_particles'], 2) interface.cleanup_code() interface.stop()
def test7(self): interface = BHTreeInterface( ) #channel_type="remote") #, debugger="xterm") interface.initialize_code() interface.commit_parameters() interface.new_particle([10, 20], [0, 0], [0, 0], [0, 0], [0, 0], [0, 0], [0, 0], [1, 1]) interface.commit_particles() retrieved_state = interface.get_state(1) self.assertEqual(10.0, retrieved_state['mass']) self.assertEqual(1, retrieved_state['radius']) retrieved_state = interface.get_state([1, 2]) self.assertEqual(20.0, retrieved_state['mass'][1]) self.assertEqual( interface.get_number_of_particles()['number_of_particles'], 2) interface.cleanup_code() interface.stop()
def test1(self): instance = BHTreeInterface() instance.initialize_code() instance.commit_parameters() 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) instance.commit_particles() 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( 1, instance.get_index_of_first_particle()['index_of_the_particle']) self.assertEquals( 2, instance.get_index_of_next_particle(1) ['index_of_the_next_particle']) instance.cleanup_code() instance.stop()
def test1(self): instance = BHTreeInterface() instance.initialize_code() instance.commit_parameters() 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) instance.commit_particles() 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(1, instance.get_index_of_first_particle()['index_of_the_particle']) self.assertEquals(2, instance.get_index_of_next_particle(1)['index_of_the_next_particle']) instance.cleanup_code() instance.stop()