Esempio n. 1
0
 def test_invalid_fen_string(self):
     """
     If fen string is not valid fen, should raise
     InvalidFENFileError.
     """
     g = GameState()
     with self.assertRaises(InvalidFENFileError):
         g._parse_fen_str('asdf 32ds 0-=fe')
Esempio n. 2
0
 def test_valid_input(self):
     """Should correctly parse a valid fen string."""
     fen = 'rnbqkbnr/pp1ppppp/8/2p5/4P3/5N2/PPPP1PPP/RNBQKB1R b KQkq - 1 2'
     g = GameState()
     g._parse_fen_str(fen)
     expected = {
         'a': {'1': 'R', '2': 'P', '3': ' ', '4': ' ', '5': ' ', '6': ' ', '7': 'p', '8': 'r'},  # noqa
         'b': {'1': 'N', '2': 'P', '3': ' ', '4': ' ', '5': ' ', '6': ' ', '7': 'p', '8': 'n'},  # noqa
         'c': {'1': 'B', '2': 'P', '3': ' ', '4': ' ', '5': 'p', '6': ' ', '7': ' ', '8': 'b'},  # noqa
         'd': {'1': 'Q', '2': 'P', '3': ' ', '4': ' ', '5': ' ', '6': ' ', '7': 'p', '8': 'q'},  # noqa
         'e': {'1': 'K', '2': ' ', '3': ' ', '4': 'P', '5': ' ', '6': ' ', '7': 'p', '8': 'k'},  # noqa
         'f': {'1': 'B', '2': 'P', '3': 'N', '4': ' ', '5': ' ', '6': ' ', '7': 'p', '8': 'b'},  # noqa
         'g': {'1': ' ', '2': 'P', '3': ' ', '4': ' ', '5': ' ', '6': ' ', '7': 'p', '8': 'n'},  # noqa
         'h': {'1': 'R', '2': 'P', '3': ' ', '4': ' ', '5': ' ', '6': ' ', '7': 'p', '8': 'r'},  # noqa
     }
     self.assertEqual(g.board, expected)
     self.assertEqual(g.player, 'b')
     self.assertTrue(g.castle_white_king)
     self.assertTrue(g.castle_white_queen)
     self.assertTrue(g.castle_black_king)
     self.assertTrue(g.castle_black_queen)
     self.assertIsNone(g.en_passant)
     self.assertEqual(g.halfmove_clock, 1)
     self.assertEqual(g.fullmove_number, 2)