def test_numbers_in_row_empty():
input_array = [
# Clear out non-zeroes in first row
[0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 1, 0, 0, 0, 0, 3, 0, 4],
[0, 0, 2, 3, 1, 5, 0, 0, 9],
[0, 0, 8, 7, 4, 6, 0, 3, 0],
[0, 7, 0, 9, 0, 1, 0, 2, 0],
[0, 9, 0, 2, 5, 3, 7, 0, 0],
[4, 0, 0, 5, 3, 8, 2, 0, 0],
[2, 0, 3, 0, 0, 0, 0, 6, 0],
[1, 5, 0, 0, 6, 0, 0, 0, 0]
]
b = Board(input_array)
assert_equals(b._numbers_in_row(0), set())
def test_numbers_in_row_valid():
input_array = [
[0, 0, 0, 0, 9, 0, 0, 5, 2],
[0, 1, 0, 0, 0, 0, 3, 0, 4],
[0, 0, 2, 3, 1, 5, 0, 0, 9],
[0, 0, 8, 7, 4, 6, 0, 3, 0],
[0, 7, 0, 9, 0, 1, 0, 2, 0],
[0, 9, 0, 2, 5, 3, 7, 0, 0],
[4, 0, 0, 5, 3, 8, 2, 0, 0],
[2, 0, 3, 0, 0, 0, 0, 6, 0],
[1, 5, 0, 0, 6, 0, 0, 0, 0]
]
b = Board(input_array)
result = b._numbers_in_row(0)
# Two sets are equal iff every element of each set is contained in the other
assert_equals(result, set((2, 5, 9)))