def test_moves_generator_black_pawn(): expected_pawn_moves = set() nose.tools.eq_(generate_all_moves(Pawn(Color.Black), 'd1'), expected_pawn_moves) expected_pawn_moves = set() expected_pawn_moves.add('a6') expected_pawn_moves.add('a5') expected_pawn_moves.add('b6') nose.tools.eq_(generate_all_moves(Pawn(Color.Black), 'a7'), expected_pawn_moves)
def test_moves_generator_white_pawn(): expected_pawn_moves = set() expected_pawn_moves.add('d3') expected_pawn_moves.add('d4') expected_pawn_moves.add('e3') expected_pawn_moves.add('c3') nose.tools.eq_(generate_all_moves(Pawn(Color.White), 'd2'), expected_pawn_moves) expected_pawn_moves = set() expected_pawn_moves.add('d8') expected_pawn_moves.add('e8') expected_pawn_moves.add('c8') nose.tools.eq_(generate_all_moves(Pawn(Color.White), 'd7'), expected_pawn_moves)
from model.Piece import Color from model.Pawn import Pawn from model.MovesGenerator import generate_all_moves generate_all_moves(Pawn(Color.White), 'd2')