def test_should_return_input_grids(self): expected = [ [0, 1], [1, 0], ] environment = Environment(expected) grids = environment.grids() self.assertEqual(grids, expected)
class EnvironmentTestCase(unittest.TestCase): def setUp(self): self.environment = Environment() def test_should_return_default_simple_grids(self): expected = [ [0, 0, 0, 0, 1], [0, 1, 0, 0, 1], [0, 0, 1, 0, 1], [0, 0, 0, 0, 1], [0, 0, 1, 0, 1], ] grids = self.environment.grids() self.assertEqual(grids, expected) def test_should_return_input_grids(self): expected = [ [0, 1], [1, 0], ] environment = Environment(expected) grids = environment.grids() self.assertEqual(grids, expected) def test_should_set_and_get_home(self): home = (1, 1) self.environment.set_home(home) self.assertEqual(self.environment.home(), home) def test_should_set_and_get_grids(self): grids = [[1, 0], [1, 0]] self.environment.set_grids(grids) self.assertEqual(self.environment.grids(), grids)