def test_make_goal_positions_9(): assert_equal( { YELLOW: frozenset([72, 73, 74, 75, 76, 77, 78, 79, 80]), GREEN: frozenset([0, 1, 2, 3, 4, 5, 6, 7, 8]), }, _make_goal_positions(9) )
def test_make_goal_positions_3(): assert_equal( {YELLOW: frozenset([6, 7, 8]), GREEN: frozenset([0, 1, 2])}, _make_goal_positions(3) )