class TestRookBlocked(unittest.TestCase): def setUp(self): self.board = Board() self.rook = Rook(self.board, 'b', '2', 'white') self.pawn = Pawn(self.board, 'b', '5', 'white') def test_blocked_moves(self): result = self.rook.get_possible_moves() self.assertEqual(len(result), 10) self.assertIn((1,0), self.rook.__moves__) self.assertIn((1,2), self.rook.__moves__) self.assertIn((1,3), self.rook.__moves__) self.assertIn((0,1), self.rook.__moves__) self.assertIn((2,1), self.rook.__moves__) self.assertIn((3,1), self.rook.__moves__) self.assertIn((4,1), self.rook.__moves__) self.assertIn((5,1), self.rook.__moves__) self.assertIn((6,1), self.rook.__moves__) self.assertIn((7,1), self.rook.__moves__)
class TestRookBase(unittest.TestCase): def setUp(self): self.board = Board() self.rook = Rook(self.board, 'b', '2', 'white') def test_base_moves(self): result = self.rook.get_possible_moves() self.assertEqual(len(result), 14) self.assertIn((1,0), self.rook.__moves__) self.assertIn((1,2), self.rook.__moves__) self.assertIn((1,3), self.rook.__moves__) self.assertIn((1,4), self.rook.__moves__) self.assertIn((1,5), self.rook.__moves__) self.assertIn((1,6), self.rook.__moves__) self.assertIn((1,7), self.rook.__moves__) self.assertIn((0,1), self.rook.__moves__) self.assertIn((2,1), self.rook.__moves__) self.assertIn((3,1), self.rook.__moves__) self.assertIn((4,1), self.rook.__moves__) self.assertIn((5,1), self.rook.__moves__) self.assertIn((6,1), self.rook.__moves__) self.assertIn((7,1), self.rook.__moves__)