berntest.py

#!/usr/bin/env python3

# Show White moves that differ between Bernstein program and the IBM 704 game

import chess as c
import chess.pgn
import bernstein as p

p.COMPC = c.WHITE
p.PLAYC = c.BLACK

pgn = open("bernstein_ibm704.pgn")
# pgn = open('bernie_16_moves_win.pgn')
game = chess.pgn.read_game(pgn)

p.PSTAB = 0
p.MAXPLIES = 3  # 4 plies

bad, badpmt = 0, 0
ml = list(game.mainline_moves())
print("pstab = %u, maxplies = %u" % (p.PSTAB, p.MAXPLIES))

# Check if correct move is in PST

b = game.board()

for m in ml:
    if b.turn == c.WHITE:
        pmt = p.get_pmt(b)
        if str(m) not in pmt:
            print(b.fullmove_number, m, pmt)
            badpmt += 1
        else:
            print("OK", b.fullmove_number, m, pmt)
    b.push(m)


print("===>", badpmt, "moves not in PMT")


print(50 * "*")

# Check if computed move differs

b = game.board()

for m in ml:
    if b.turn == c.WHITE:
        t, r = p.getmove(b, usebook=False, silent=False)
        if r[0] != str(m):
            print(b.fullmove_number, m, r[0], 30 * "<")
            bad += 1
    b.push(m)

print("===>", bad, "moves differ (%u%%)" % (200 * bad / len(ml)))