class TestGrid(unittest.TestCase):
''' Test Grid object '''
def setUp(self):
self.size = 200, 100
self.grid = Grid(self.size)
def test_add(self):
content = Content()
pos = 1, 2
size = 2, 1
cells = set([(1, 2), (2, 2)])
self.grid.add(content, pos, size)
self.assertEqual(cells, self.grid._contents[content])
self.assertEqual([content, ], list(self.grid._grid[1, 2]))
self.assertEqual([content, ], list(self.grid._grid[2, 2]))
def test_remove(self):
content = Content()
pos = 1, 2
size = 2, 1
self.grid.add(content, pos, size)
self.grid.remove(content)
self.assertNotIn(content, self.grid._contents)
cells = set([(1, 2), (2, 2)])
for cell in cells:
self.assertNotIn(cell, self.grid._grid)
def test_get(self):
content = Content()
pos = 1, 2
size = 2, 1
self.grid.add(content, pos, size)
grid_content = self.grid.get((0, 0), self.size)
self.assertIn(content, grid_content)
self.assertEqual(1, len(grid_content))
def setUp(self):
self.size = 200, 100
self.grid = Grid(self.size)
