def test_remove_id_end(self): buffer = ShapeBuffer('shape', 4) buffer.new(0, MODEL_ONES, COLOR_ONES) buffer.new(1, MODEL_TWOS, COLOR_TWOS) buffer.new(2, MODEL_THREES, COLOR_THREES) self.assert_turtle_data(buffer, 2, 2, MODEL_THREES, COLOR_THREES) buffer.remove(2) self.assertEqual(buffer.count, 2) self.assert_turtle_data(buffer, 0, 0, MODEL_ONES, COLOR_ONES) self.assert_turtle_data(buffer, 1, 1, MODEL_TWOS, COLOR_TWOS) # check last one zeroed self.assert_turtle_data(buffer, None, 2, MODEL_ZEROS, COLOR_ZEROS) self.assertNotIn(2, buffer.id_to_index) self.assertNotIn(2, buffer.index_to_id)