def test_valid_colours(): state = SpireState() expected_colours = (state.BLACK, state.RED) colours = state.get_valid_colours() assert colours == expected_colours
def test_valid_colours_no_red(): state = SpireState("""\ A C E G 7 . . . . 7 5 . . . . 5 3 . . . . 3 1 R . . . 1 A C E G >B """) expected_colours = (state.BLACK, ) colours = state.get_valid_colours() assert colours == expected_colours