def test_set_cell(self): world = World(4, 0) expected_output = ArableLandCell(world, [0, 0], 0) world.set_cell(0, 0, expected_output) submited_output = world.get_cell(0, 0) self.assertEqual(submited_output, expected_output)
def test_get_cell(self): world = World(4, 0) cell = world.get_cell(0, 0) self.assertEqual(type(cell), LandCell)