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()