add unit test for n-queens

C++/CMakeLists.txt

@@ -19,6 +19,15 @@ add_library(test_runner STATIC
 # Algorithms
 # ============================================================================
 
+# -------------------
+# Backtracking
+# -------------------
+
+# N-Queens
+add_executable(n_queens
+        test/algorithm/backtracking/n_queens.cpp)
+target_link_libraries(n_queens test_runner)
+
 # -------------------
 # Dynamic programming
 # -------------------

C++/test/algorithm/backtracking/n_queens.cpp

@@ -0,0 +1,49 @@
+#include "third_party/catch.hpp"
+#include "algorithm/backtracking/n_queens.hpp"
+
+TEST_CASE("Unplaceable cases", "[backtracking][n_queens]") {
+    NQueensSolver b0(0);
+    REQUIRE_FALSE(b0.can_place_queens());
+    NQueensSolver b2(2);
+    REQUIRE_FALSE(b2.can_place_queens());
+    NQueensSolver b3(3);
+    REQUIRE_FALSE(b3.can_place_queens());
+}
+
+TEST_CASE("Placeable cases", "[backtracking][n_queens]") {
+    NQueensSolver b1(1);
+    REQUIRE(b1.can_place_queens());
+    REQUIRE(b1.num_solutions() == 1);
+    Board b1_sln {{true}};
+    REQUIRE(b1.get_solution() == b1_sln);
+    std::vector<Board> b1_slns {{{true}}};
+    REQUIRE(b1.get_solutions() == b1_slns);
+
+    NQueensSolver b4(4);
+    REQUIRE(b4.can_place_queens());
+    REQUIRE(b4.num_solutions() == 2);
+    Board b4_sln1 {
+        {false, true, false, false},
+        {false, false, false, true},
+        {true, false, false, false},
+        {false, false, true, false}
+    };
+    Board b4_sln2 {
+        {false, false, true, false},
+        {true, false, false, false},
+        {false, false, false, true},
+        {false, true, false, false}
+    };
+    using Catch::Matchers::VectorContains;
+    REQUIRE_THAT(b4.get_solutions(), VectorContains(b4_sln1));
+    REQUIRE_THAT(b4.get_solutions(), VectorContains(b4_sln2));
+
+    NQueensSolver b9(9);
+    REQUIRE(b9.can_place_queens());
+    REQUIRE(b9.num_solutions() == 352);
+
+    NQueensSolver b11(11);
+    REQUIRE(b11.can_place_queens());
+    REQUIRE(b11.num_solutions() == 2680);
+
+}