ChessBot/tests.py at main · TJ5/ChessBot · 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
91
92
93
94
95
96
97
98
99
100
101
from endgame import EndgamePredictor
import unittest
from game import Game
import chess
from mockclient import MockClient
from bot import Bot
from value import SquareValue
from movetree import MoveTreeNode
from boardwrapper import BoardWrapper
from table import TTable
import math
import threading
event = threading.Event()
class TestBot(unittest.TestCase):
    def setUp(self) -> None:

        self.bot = Bot("black")
        self.value = SquareValue()
        self.timer = threading.Timer(15, self.flag)
    def flag(self):
        event.set()
    def test_game_init(self):
        m = MockClient()
        game = Game(m, True)


        game.join()
        current_state = game.current_state
        self.assertEqual(current_state["type"], "gameFull")
        self.assertEqual(current_state["white"]["id"], "whenchess")

        self.assertTrue(m.moves[0])
        self.assertTrue(m.moves[1])
        self.assertEqual(m.moves[2], "c3d5")

    def test_bot_movemaking(self):
        self.bot.updateboardall("e2e4 e7e5")
        move = self.bot.getmove()
        self.assertFalse(move)
        self.bot.updateboardall("e2e4 e7e5 g1f3")
        move = self.bot.getmove()
        testboard = chess.Board("rnbqkbnr/pppp1ppp/8/4p3/4P3/5N2/PPPP1PPP/RNBQKB1R b KQkq - 1 2")
        self.assertTrue(chess.Move.from_uci(move) in testboard.legal_moves)


    def test_pawn_eval(self):
        self.assertEqual(self.value.getpawnmodifier(chess.E4, chess.WHITE), 20)
        self.assertEqual(self.value.getpawnmodifier(chess.E2, chess.BLACK), 50)

        self.bot.updateboardall("e2e4")
        self.assertEqual(self.bot.shalloweval(), -40)
    def test_knight_eval(self):
        self.bot.updateboardall("e2e4 g8f6")
        self.assertEqual(self.bot.shalloweval(), 10)
    def test_bishop_eval(self):
        self.bot.updateboardall("e2e4 e7e5 f1c4")
        self.assertEqual(self.bot.shalloweval(), -20)


    def test_table(self):
        t = TTable()
        e = EndgamePredictor()
        b = BoardWrapper(e)
        self.timer.start()
        tree = MoveTreeNode(b, 0, 4, chess.WHITE, e)
        bestboard = tree.addchildren(event, (-1 * math.inf), math.inf)
        t.put(b.board,bestboard.getmovestack(), 4)
        self.assertEqual(t.table[5060803636482931868], ['rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR', [chess.Move.from_uci('g1f3'), chess.Move.from_uci('g8f6'), chess.Move.from_uci('b1c3'), chess.Move.from_uci('b8c6')], 4, chess.WHITE])
        self.assertEqual(t.get(b.board)[1][0], chess.Move.from_uci('g1f3'))


    def test_tree_table_lookup(self):

        e = EndgamePredictor()
        b = BoardWrapper(e)
        tree = MoveTreeNode(b, 0, 4, chess.WHITE, e)
        self.timer.start()
        tree.getbestmove(event)
        s = tree.size()
        print(str(s))
        #self.assertLess(s, 3346)

    def test_MVVLVA(self):
        e = EndgamePredictor()
        bw = BoardWrapper(e, chess.Board("r1bqk2r/1ppp1p1p/p1n3pn/2b1p1B1/1PB1P3/3P1N2/P1P2PPP/RN1QK2R w KQkq - 0 7"))
        moves = bw.getsortedmoves()

        self.assertEqual(moves[0], chess.Move.from_uci('g5d8'))

    def test_hash_move_ordering(self):
        e = EndgamePredictor()
        bw = BoardWrapper(e, chess.Board())
        moves = bw.getsortedmoves(chess.Move.from_uci('e2e4'))
        self.assertEqual(moves[0], chess.Move.from_uci('e2e4'))
    def test_endgame(self):
        e = EndgamePredictor()
        bw = BoardWrapper(e, chess.Board("rnbqkbnr/pp2pppp/3p4/2p5/4P3/5N2/PPPP1PPP/RNBQKB1R w KQkq - 0 3"))
        self.assertFalse(bw.is_endgame())
        bw = BoardWrapper(e, chess.Board("8/5p2/2r5/8/p2KP3/P7/6kp/3R4 w - - 8 70"))
        self.assertTrue(bw.is_endgame())
unittest.main()