def test_generate_correct_sq_piece_data():
state_gen = ExpertStateGenerator("bogus", 'csv', 8,
"yureka/tests/test.pgn", 1)
g = next(state_gen.get_game())
df = pd.DataFrame(state_gen.get_game_data(g))
assert set(df.loc[0, 'white_square_piece_0'].split(',')) == set(
('a1-R,b1-N,c1-B,d1-Q,e1-K,f1-B,g1-N,h1-R,a2-P,b2-P,c2-P,'
'd2-P,e2-P,f2-P,g2-P,h2-P').split(','))
assert set(df.loc[0, 'black_square_piece_0'].split(',')) == set(
('a7-p,b7-p,c7-p,d7-p,e7-p,f7-p,g7-p,h7-p,a8-r,b8-n,c8-b,'
'd8-q,e8-k,f8-b,g8-n,h8-r').split(','))
for i in range(1, 8):
assert df.loc[0, 'white_square_piece_%d' % i] == ''
assert df.loc[0, 'black_square_piece_%d' % i] == ''
assert set(df.loc[1, 'white_square_piece_0'].split(',')) == set(
('h8-R,g8-N,f8-B,e8-Q,d8-K,c8-B,b8-N,a8-R,h7-P,g7-P,f7-P,'
'e7-P,c7-P,b7-P,a7-P,d5-P').split(','))
assert set(df.loc[1, 'black_square_piece_0'].split(',')) == set(
('h2-p,g2-p,f2-p,e2-p,d2-p,c2-p,b2-p,a2-p,h1-r,g1-n,f1-b,'
'e1-q,d1-k,c1-b,b1-n,a1-r').split(','))
assert set(df.loc[1, 'white_square_piece_1'].split(',')) == set(
('h8-R,g8-N,f8-B,e8-Q,d8-K,c8-B,b8-N,a8-R,h7-P,g7-P,f7-P,'
'e7-P,c7-P,b7-P,a7-P,d7-P').split(','))
assert set(df.loc[1, 'black_square_piece_1'].split(',')) == set(
('h2-p,g2-p,f2-p,e2-p,d2-p,c2-p,b2-p,a2-p,h1-r,g1-n,f1-b,'
'e1-q,d1-k,c1-b,b1-n,a1-r').split(','))
for i in range(2, 8):
assert df.loc[0, 'white_square_piece_%d' % i] == ''
assert df.loc[0, 'black_square_piece_%d' % i] == ''
def test_move_count_data():
state_gen = ExpertStateGenerator("bogus", 'csv', 8,
"yureka/tests/test.pgn", 1)
game = next(state_gen.get_game())
df = pd.DataFrame(state_gen.get_game_data(game))
for i, data in df.iterrows():
assert int(i / 2) + 1 == data['move_count']
def test_repetition_data():
b = chess.Board(fen='4k3/8/8/8/8/8/8/4K3 w - - 0 1')
# Create transpositions
def move(b):
b.push(chess.Move.from_uci('e1e2'))
b.push(chess.Move.from_uci('e8e7'))
b.push(chess.Move.from_uci('e2e1'))
b.push(chess.Move.from_uci('e7e8'))
for i in range(3):
move(b)
game = chess.pgn.Game.from_board(b)
state_gen = ExpertStateGenerator("bogus", 'csv', 8,
"yureka/tests/test.pgn", 1)
df = pd.DataFrame(state_gen.get_game_data(game))
for i, data in df.iterrows():
if i < 4:
for j in range(8):
# no transpositions yet
assert data['rep_2_%d' % j] == 0
assert data['rep_3_%d' % j] == 0
elif i < 8:
# twofold transpositions
for j in range(i - 3):
assert data['rep_2_%d' % j] == 1
for j in range(i - 3, 8):
assert data['rep_2_%d' % j] == 0
for j in range(8):
assert data['rep_3_%d' % j] == 0
else:
# threefold transpositions
for j in range(i - 3):
assert data['rep_2_%d' % j] == 1
for j in range(i - 3, 8):
assert data['rep_2_%d' % j] == 0
for j in range(i - 7):
assert data['rep_3_%d' % j] == 1
for j in range(i - 7, 8):
assert data['rep_3_%d' % j] == 0
def test_no_progress_count_data():
b = chess.Board(fen='4k3/8/8/8/8/8/8/4K3 w - - 0 1')
# no progress moves
def move(b):
b.push(chess.Move.from_uci('e1e2'))
b.push(chess.Move.from_uci('e8e7'))
b.push(chess.Move.from_uci('e2e1'))
b.push(chess.Move.from_uci('e7e8'))
for i in range(50):
move(b)
game = chess.pgn.Game.from_board(b)
state_gen = ExpertStateGenerator("bogus", 'csv', 8,
"yureka/tests/test.pgn", 1)
df = pd.DataFrame(state_gen.get_game_data(game))
for i, data in df.iterrows():
assert data['no_progress'] == int(i / 2)
def test_castling_data():
state_gen = ExpertStateGenerator("bogus", 'csv', 8,
"yureka/tests/test.pgn", 1)
def get_castling_game(king_side=True):
b = chess.Board(fen='r3k2r/8/8/8/8/8/8/R3K2R w - - 0 1')
# hack to reset the castling rights
b.castling_rights = chess.BB_CORNERS
if king_side:
b.push(chess.Move.from_uci('e1g1'))
b.push(chess.Move.from_uci('e8g8'))
b.push(chess.Move.from_uci('g1g2'))
else:
b.push(chess.Move.from_uci('e1c1'))
b.push(chess.Move.from_uci('e8c8'))
b.push(chess.Move.from_uci('c1c2'))
return chess.pgn.Game.from_board(b)
test_cases = [
{
'name':
'Kingside',
'game':
get_castling_game(),
'expected_data':
pd.DataFrame([
{
'w_kingside_castling': 1,
'w_queenside_castling': 1,
'b_kingside_castling': 1,
'b_queenside_castling': 1,
},
{
'w_kingside_castling': 0,
'w_queenside_castling': 0,
'b_kingside_castling': 1,
'b_queenside_castling': 1,
},
{
'w_kingside_castling': 0,
'w_queenside_castling': 0,
'b_kingside_castling': 0,
'b_queenside_castling': 0,
},
]),
},
{
'name':
'Queenside',
'game':
get_castling_game(king_side=False),
'expected_data':
pd.DataFrame([
{
'w_kingside_castling': 1,
'w_queenside_castling': 1,
'b_kingside_castling': 1,
'b_queenside_castling': 1,
},
{
'w_kingside_castling': 0,
'w_queenside_castling': 0,
'b_kingside_castling': 1,
'b_queenside_castling': 1,
},
{
'w_kingside_castling': 0,
'w_queenside_castling': 0,
'b_kingside_castling': 0,
'b_queenside_castling': 0,
},
]),
},
]
for tc in test_cases:
df = pd.DataFrame(state_gen.get_game_data(tc['game']))
columns = [
'b_kingside_castling',
'b_queenside_castling',
'w_kingside_castling',
'w_queenside_castling',
]
assert df[columns].equals(tc['expected_data'])