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