class TestTable(unittest.TestCase): def setUp(self): self.table1 = Table('table1', 5, 5) self.table2 = Table('table1', 10, 10) self.table3 = Table('table1', 50, 50, {'TO_SOUTH': 1, 'TO_NORTH': -1, 'TO_WEST': 1, 'TO_EAST': -1}) self.table4 = Table('table1', 100, 100, {'TO_SOUTH': 1, 'TO_NORTH': -1, 'TO_WEST': -1, 'TO_EAST': 1}) def test_get_coordinate(self): self.assertTrue(Coordinate(3, 3) == self.table1.get_coordinate(3, 3)) self.assertTrue(Coordinate(8, 9) == self.table2.get_coordinate(8, 9)) self.assertTrue(Coordinate(34, 45) == self.table3.get_coordinate(34, 45)) self.assertTrue(Coordinate(78, 88) == self.table4.get_coordinate(78, 88))