CheckPuzzlesTb/test.py at master · kraktus/CheckPuzzlesTb · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#!/usr/local/bin/python3
#coding: utf-8

"""
Confront lichess puzzles V2 to syzygy tablebase
"""

import unittest

from t import PuzzleChecker, Puzzle, Error

#########
#Classes#
#########

class Test(unittest.TestCase):

    def setUp(self):
        self.checker = PuzzleChecker()

    def test_wrong_winning_puzzle(self):
        puzzle = Puzzle(fen="8/2R5/1P4p1/2K5/5pk1/5b2/8/8 w - - 0 50", # id (now deletd): 0XVra
            moves="c7c6 f3c6 c5c6 f4f3 b6b7 f3f2 b7b8q f2f1q".split(),
            expected_winning=True)
        self.assertEqual(self.checker.check_puzzle(puzzle), set([Error.Wrong]))

    def test_multiple_right_moves_winning_puzzle(self):
        puzzle = Puzzle(fen="8/8/8/P6R/5K2/1r5p/5k2/8 w - - 3 57", # id (now deletd): 0BFLx
            moves="a5a6 b3b4 f4g5 b4b5 g5g6 b5h5 g6h5 h3h2".split(),
            expected_winning=True)
        self.assertEqual(self.checker.check_puzzle(puzzle), set([Error.Multiple]))


    def test_wrong_multiple_drawing_puzzle(self):
        puzzle = Puzzle(fen="8/8/8/K1Pk1pp1/8/7P/1P6/8 w - - 0 43", # id (in fact winning): 05dOY
            moves="b2b4 g5g4 h3g4 f5g4 b4b5 g4g3 c5c6 d5d6".split(),
            expected_winning=False)
        self.assertEqual(self.checker.check_puzzle(puzzle), set([Error.Wrong, Error.Multiple]))

    def test_right_mate_puzzle_with_multiple_mates(self):
        puzzle = Puzzle(fen="1k6/2Q5/1K6/8/8/8/8/4qq2 b - - 0 1",
            moves="b8a8 c7c8".split(),
            expected_winning=True,
            mate=2)
        self.assertEqual(self.checker.check_puzzle(puzzle), set())

    def test_right_mate_without_DTM(self):
        puzzle = Puzzle(fen="1k6/8/1KR5/7q/7q/8/8/5q1q b - - 0 1",
            moves="b8a8 c6c8".split(),
            expected_winning=True,
            mate=2)
        self.assertEqual(self.checker.check_puzzle(puzzle), set())

    def test_wrong_mate_with_DTM(self):
        puzzle = Puzzle(fen="1k6/8/1K6/8/8/8/1R6/8 b - - 0 1",
            moves="b8a8 b2d2 a8b8 d2d8".split(),
            expected_winning=True,
            mate=4)
        self.assertEqual(self.checker.check_puzzle(puzzle), set([Error.Multiple]))

    def test_right_mate_puzzle_with_multiple_solution_DTZ_only(self):
        puzzle = Puzzle(fen="1k6/7Q/1K6/8/8/8/6nn/6nn b - - 0 1",
            moves="b8a8 h7h8".split(),
            expected_winning=True,
            mate=2)
        self.assertEqual(self.checker.check_puzzle(puzzle), set())

    def test_right_mate_puzzle_DTZ_only(self):
        """There is one mate in 2 and mate in 3
        wrong puzzle but not able to detect it for now"""
        puzzle = Puzzle(fen="1k6/8/1K6/8/8/8/1R3nn1/6nn b - - 3 3",
            moves="b8a8 b6c7 a8a7 b2a2".split(),
            expected_winning=True,
            mate=4)
        self.assertEqual(self.checker.check_puzzle(puzzle), set())

    def test_not_mate_puzzle(self):
        puzzle = Puzzle(fen="1k6/2Q5/1K6/8/8/8/8/4qq2 b - - 0 1",
            moves="b8a8 c7b8".split(), # giving up the queen
            expected_winning=True,
            mate=2)
        self.assertEqual(self.checker.check_puzzle(puzzle), set([Error.Wrong]))

######
#Main#
######

if __name__ == "__main__":
    print("#"*80)
    unittest.main()