def testTotalNQueens(self): s = Solution() ns = [1, 2, 3, 4] ans = [1, 0, 0, 2] for n, an in zip(ns, ans): num = s.totalNQueens(n) self.assertEqual(num, an)
from NQueensII import Solution s = Solution() print s.totalNQueens(8)