def test_destroy_turtle(self):
manager = BufferManager(4)
model, color = manager.create_turtle(0, 'classic', MODEL_ONES,
COLOR_ONES)
manager.destroy_turtle(0)
self.assertEqual(list(model), MODEL_ZEROS)
self.assertEqual(list(color), COLOR_ZEROS)
self.assertEqual(manager.buffers['classic'].count, 0)
def test_set_shape(self):
manager = BufferManager(4)
model, color = manager.create_turtle(0, 'classic', MODEL_ONES,
COLOR_ONES)
model2, color2 = manager.set_shape(0, 'turtle')
self.assertEqual(len(manager.buffers), 2)
self.assertIn('turtle', manager.buffers)
self.assertEqual(list(model2), MODEL_ONES)
self.assertEqual(list(color2), COLOR_ONES)
def test_create_turtle(self):
manager = BufferManager(4)
model, color = manager.create_turtle(0, 'classic', MODEL_ONES,
COLOR_ONES)
self.assertEqual(list(model), MODEL_ONES)
self.assertEqual(list(color), COLOR_ONES)
self.assertEqual(len(manager.buffers), 1)
self.assertIn('classic', manager.buffers)
self.assertEqual(manager.buffers['classic'].size, 4)
def test_get_buffer(self):
manager = BufferManager(4)
self.assertEqual(len(manager.buffers), 0)
buffer1 = manager.get_buffer('classic')
self.assertEqual(len(manager.buffers), 1)
self.assertIn('classic', manager.buffers)
self.assertEqual(buffer1.size, 4)
buffer2 = manager.get_buffer('classic')
self.assertEqual(len(manager.buffers), 1)
self.assertIs(buffer1, buffer2)
def __init__(self, width, height, samples=None, buffer_size=16):
self.width = width
self.half_width = width // 2
self.height = height
self.half_height = height // 2
self.create_window(width, height, samples)
self.set_background_color()
self.compile_program()
self.setup_vaos()
self.manager = BufferManager(buffer_size)
self.perspective_matrix = identity()
self.set_perspective()
self.view_matrix = identity()
self.view_matrix[12] = 0.0
self.view_matrix[13] = 0.0
self.view_matrix[14] = 0.0
self.set_view()
