Exemplo n.º 1
0
def test_route_one_one_stop():
    g = Graph()
    pandora_v = g.add('pandora')
    arendelle_v = g.add('arendelle')
    metroville_v = g.add('metroville')
    monstropolus_v = g.add('monstropolus')
    naboo_v = g.add('naboo')
    narnia_v = g.add('narnia')
    g.add_edge(pandora_v, arendelle_v, 150)
    g.add_edge(pandora_v, metroville_v, 82)
    g.add_edge(arendelle_v, metroville_v, 99)
    g.add_edge(arendelle_v, monstropolus_v, 42)
    g.add_edge(naboo_v, monstropolus_v, 73)
    g.add_edge(metroville_v, monstropolus_v, 105)
    g.add_edge(naboo_v, metroville_v, 26)
    g.add_edge(naboo_v, narnia_v, 250)
    g.add_edge(narnia_v, metroville_v, 37)
    actual = get_edges(g, ['naboo', 'metroville', 'arendelle'])
    expected = (True, 125)
    assert actual == expected
Exemplo n.º 2
0
def test_fail_one():
    g = Graph()
    pandora_v = g.add('pandora')
    arendelle_v = g.add('arendelle')
    metroville_v = g.add('metroville')
    monstropolus_v = g.add('monstropolus')
    naboo_v = g.add('naboo')
    narnia_v = g.add('narnia')
    g.add_edge(pandora_v, arendelle_v, 150)
    g.add_edge(pandora_v, metroville_v, 82)
    g.add_edge(arendelle_v, metroville_v, 99)
    g.add_edge(arendelle_v, monstropolus_v, 42)
    g.add_edge(naboo_v, monstropolus_v, 73)
    g.add_edge(metroville_v, monstropolus_v, 105)
    g.add_edge(naboo_v, metroville_v, 26)
    g.add_edge(naboo_v, narnia_v, 250)
    g.add_edge(narnia_v, metroville_v, 37)
    actual = get_edges(g, ['pandora', 'naboo', 'narnia', 'naboo'])
    expected = (False, 0)
    assert actual == expected