def test_find_ways_one_way(): nodes = Nodes(3) nodes.connect(0, 1) nodes.connect(1, 2) ways = nodes.find_ways(nodes[0]) assert [[0, 1, 2]] == ways
def test_find_ways(): nodes = Nodes(4) nodes.connect(0, 1) nodes.connect(1, 2) nodes.connect(1, 3) nodes.connect(2, 3) nodes.connect(3, 0) ways = nodes.find_ways(nodes[0]) assert [0, 1, 2, 3, 0] in ways assert [0, 3, 2, 1, 0] in ways
def test_find_ways_1(): nodes = Nodes(3) nodes.connect(0, 1) nodes.connect(1, 2) nodes.connect(2, 0) ways = nodes.find_ways(nodes[0]) assert 2 == len(ways) assert [ [0, 1, 2, 0], [0, 2, 1, 0], ] == ways
def test_good_run(): nodes = Nodes(4) nodes.connect(0, 1) nodes.connect(1, 2) nodes.connect(1, 3) nodes.connect(2, 3) nodes.connect(3, 0) ways = nodes.get_good_ways() print(ways) assert 8 == len(ways) ways = nodes.find_ways(nodes[1]) print(ways)