def test2(self): instance = BHTreeInterface() instance.initialize_code() instance.commit_parameters() for i in [1, 2, 3]: 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.commit_particles() 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']) self.assertEquals(3, instance.get_index_of_next_particle(2)['index_of_the_next_particle']) instance.delete_particle(1) self.assertEquals(2, instance.get_number_of_particles()['number_of_particles']) #the deletion does a swap, so 3 is copied to 1, (overwriting old 1 and treesize -> treesize-1 self.assertEquals(3, instance.get_index_of_first_particle()['index_of_the_particle']) self.assertEquals(1, instance.get_index_of_next_particle(2)['__result']) instance.cleanup_code() instance.stop()
def test6(self): instance = BHTreeInterface() instance.initialize_code() instance.commit_parameters() ids = [] for i in [1, 2, 3]: id, error = 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) ids.append(id) print ids instance.commit_particles() instance.delete_particle(ids[0]) id, error = instance.new_particle(mass=4, radius=1.0, x=0.0, y=0.0, z=0.0, vx=0.0, vy=0.0, vz=0.0) self.failIfEqual(id, ids[-1]) instance.cleanup_code() instance.stop()
def test2(self): instance = BHTreeInterface() instance.initialize_code() instance.commit_parameters() for i in [1, 2, 3]: 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.commit_particles() 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']) self.assertEquals( 3, instance.get_index_of_next_particle(2) ['index_of_the_next_particle']) instance.delete_particle(1) self.assertEquals( 2, instance.get_number_of_particles()['number_of_particles']) #the deletion does a swap, so 3 is copied to 1, (overwriting old 1 and treesize -> treesize-1 self.assertEquals( 3, instance.get_index_of_first_particle()['index_of_the_particle']) self.assertEquals(1, instance.get_index_of_next_particle(2)['__result']) instance.cleanup_code() instance.stop()
def test6(self): instance = BHTreeInterface() instance.initialize_code() instance.commit_parameters() ids = [] for i in [1, 2, 3]: id, error = 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) ids.append(id) print ids instance.commit_particles() instance.delete_particle(ids[0]) id, error = instance.new_particle(mass = 4, radius = 1.0, x = 0.0, y = 0.0, z = 0.0, vx = 0.0, vy = 0.0, vz = 0.0) self.failIfEqual(id, ids[-1]) instance.cleanup_code() instance.stop()