test case for zebra problem

dsaw · dsaw · commit 34997e48547a · 2018-03-13T00:11:41.000+05:30
diff --git a/tests/test_csp.py b/tests/test_csp.py
@@ -382,6 +382,8 @@ def test_tree_csp_solver():
     assert (tcs['NT'] == 'R' and tcs['WA'] == 'B' and tcs['Q'] == 'B' and tcs['NSW'] == 'R' and tcs['V'] == 'B') or \
            (tcs['NT'] == 'B' and tcs['WA'] == 'R' and tcs['Q'] == 'R' and tcs['NSW'] == 'B' and tcs['V'] == 'R')
 
-
+def test_zebra_problem():
+    zebra_solution = solve_zebra()
+    assert(zebra_solution[0] == 5 and zebra_solution[1] == 1)
 if __name__ == "__main__":
     pytest.main()
