test/unit/test_perft.cpp

   1 #include "../../src/common.h"
   2 #include "../../src/game.h"
   3 #include "gtest/gtest.h"
   4
   5 class PerftTest : public testing::Test
   6 {
   7     protected:
   8         Game game;
   9         std::string fen;
  10 };
  11
  12 TEST_F(PerftTest, StartPosition)
  13 {
  14     fen = "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1";
  15     game.init(fen);
  16     EXPECT_EQ(20, game.perft(1));
  17     EXPECT_EQ(400, game.perft(2));
  18
  19     // Test captures and checks
  20     EXPECT_EQ(8902, game.perft(3));
  21
  22     // Test checkmates
  23     EXPECT_EQ(197281, game.perft(4));
  24
  25     // Test en passants
  26     //EXPECT_EQ(4865609, game.perft(5));
  27 }
  28
  29 TEST_F(PerftTest, CastlePosition)
  30 {
  31     fen = "r3k2r/p1ppqpb1/bn2pnp1/3PN3/1p2P3/2N2Q1p/PPPBBPPP/R3K2R w KQkq -";
  32     game.init(fen);
  33     EXPECT_EQ(48, game.perft(1));
  34     EXPECT_EQ(2039, game.perft(2));
  35     EXPECT_EQ(97862, game.perft(3));
  36 }
  37
  38 TEST_F(PerftTest, CheckPosition)
  39 {
  40     fen = "8/2p5/3p4/KP5r/1R3p1k/8/4P1P1/8 w - -";
  41     game.init(fen);
  42     EXPECT_EQ(14, game.perft(1));
  43     EXPECT_EQ(191, game.perft(2));
  44     EXPECT_EQ(2812, game.perft(3));
  45     EXPECT_EQ(43238, game.perft(4));
  46 }
  47
  48 TEST_F(PerftTest, Promotion1Position)
  49 {
  50     fen = "r3k2r/Pppp1ppp/1b3nbN/nP6/BBP1P3/q4N2/Pp1P2PP/R2Q1RK1 w kq - 0 1";
  51     game.init(fen);
  52     EXPECT_EQ(6, game.perft(1));
  53     EXPECT_EQ(264, game.perft(2));
  54     EXPECT_EQ(9467, game.perft(3));
  55     EXPECT_EQ(422333, game.perft(4));
  56 }
  57
  58 TEST_F(PerftTest, Promotion2Position)
  59 {
  60     fen = "r2q1rk1/pP1p2pp/Q4n2/bbp1p3/Np6/1B3NBn/pPPP1PPP/R3K2R b KQ - 0 1";
  61     game.init(fen);
  62     EXPECT_EQ(6, game.perft(1));
  63     EXPECT_EQ(264, game.perft(2));
  64     EXPECT_EQ(9467, game.perft(3));
  65     EXPECT_EQ(422333, game.perft(4));
  66 }
  67
  68 TEST_F(PerftTest, Promotion3Position)
  69 {
  70     fen = "n1n5/PPPk4/8/8/8/8/4Kppp/5N1N b - - 0 1";
  71     game.init(fen);
  72     EXPECT_EQ(24, game.perft(1));
  73     EXPECT_EQ(496, game.perft(2));
  74     EXPECT_EQ(9483, game.perft(3));
  75     EXPECT_EQ(182838, game.perft(4));
  76 }
  77
  78 TEST_F(PerftTest, CaptureBug1Position)
  79 {
  80     fen = "r6R/4k3/8/8/8/8/8/4K2R b K - 0 2";
  81     game.init(fen);
  82     EXPECT_EQ(19, game.perft(1));
  83     EXPECT_EQ(447, game.perft(2)); // Was 446 before bug fix #59cef02
  84     EXPECT_EQ(7518, game.perft(3));
  85     //EXPECT_EQ(176792, game.perft(4));
  86 }
  87
  88 TEST_F(PerftTest, MaxMoves1Position)
  89 {
  90     fen = "3Q4/1Q4Q1/4Q3/2Q4R/Q4Q2/3Q4/1Q4Rp/1K1BBNNk w - - 0 1";
  91     game.init(fen);
  92     EXPECT_EQ(218, game.perft(1)); // Need MAX_PLY > 218
  93     EXPECT_EQ(20, game.perft(2));
  94     EXPECT_EQ(4072, game.perft(3));
  95     EXPECT_EQ(15903, game.perft(4));
  96 }
  97
  98 TEST_F(PerftTest, MaxMoves2Position)
  99 {
 100     fen = "n1r1r1b1/1P1P1P1P/1Q6/3NBNK1/R7/4p1p1/3PBPkP/2R5 w - - 0 1";
 101     game.init(fen);
 102     EXPECT_EQ(144, game.perft(1));
 103     EXPECT_EQ(3283, game.perft(2));
 104     EXPECT_EQ(407751, game.perft(3));
 105 }
 106
 107 TEST_F(PerftTest, MaxCapturesPosition)
 108 {
 109     fen = "r1n1n1b1/1P1P1P1P/1N1N1N2/2RnQrRq/2pKp3/3BNQbQ/k7/4Bq2 w - - 0 1";
 110     game.init(fen);
 111     EXPECT_EQ(127, game.perft(1));
 112     EXPECT_EQ(6926, game.perft(2));
 113     EXPECT_EQ(767038, game.perft(3));
 114 }