def test_get_number_of_queens_on_board(self): size = 5 queens = (XYLocation(1, 1), XYLocation(2, 4), XYLocation(3, 3)) nqb = NQueensBoard(size) nqb.set_board(queens) self.assertEqual(3, nqb.get_number_of_queens_on_board())