class Test(unittest.TestCase): def setUp(self): self.trans = Translation() def test_if_translate_to_left_shift_all_positions_to_left(self): new_points = self.trans.in_direction( [Point(4, 2), Point(1, 5)], LEFT) self.check_point(4, 1, new_points[0]) self.check_point(1, 4, new_points[1]) def test_if_translate_to_right_then_shift_all_positions_to_right(self): new_points = self.trans.in_direction( [Point(4, 2), Point(1, 5)], RIGHT) self.check_point(4, 3, new_points[0]) self.check_point(1, 6, new_points[1]) def test_if_translate_down_then_all_points_go_down(self): new_points = self.trans.in_direction( [Point(4, 2), Point(1, 5)], DOWN) self.check_point(5, 2, new_points[0]) self.check_point(2, 5, new_points[1]) def check_point(self, row, col, point): self.assertEqual(row, point.row) self.assertEqual(col, point.col)
def setUp(self): self.trans = Translation()