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)