def test_mutlple_new(self): buffer = ShapeBuffer('shape', 4) buffer.new(0) self.assert_id_map(buffer, 0, 0) self.assertEqual(buffer.count, 1) buffer.new(1) self.assert_id_map(buffer, 1, 1) self.assertEqual(buffer.count, 2)
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)
def test_new_with_init(self): buffer = ShapeBuffer('shape', 4) model, color = buffer.new(0, MODEL_ONES, COLOR_TWOS) self.assert_turtle_data(buffer, 0, 0, MODEL_ONES, COLOR_TWOS)
def test_new_bad_id(self): buffer = ShapeBuffer('shape', 4) buffer.new(0) with self.assertRaises(AssertionError): buffer.new(0)
def test_new(self): buffer = ShapeBuffer('shape', 4) model, color = buffer.new(0) self.assert_turtle_data(buffer, 0, 0, MODEL_ZEROS, COLOR_ZEROS) self.assertEqual(buffer.count, 1)