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)