Ejemplo n.º 1
0
def test_anyhashableg_as_weighted_with_caching_and_write_throught_with_unweighted(
):
    g = create_graph(
        directed=False,
        allowing_self_loops=True,
        allowing_multiple_edges=False,
        weighted=False,
        any_hashable=True,
    )

    g.add_vertex(0)
    g.add_vertex(1)
    g.add_edge(0, 1, edge=0)

    with pytest.raises(ValueError):
        g.set_edge_weight(0, 5.0)

    with pytest.raises(ValueError):
        g.edge_attrs[0]['weight'] = 5.0

    assert g.get_edge_weight(0) == 1.0

    def edge_weight(e):
        return 100.5

    with pytest.raises(ValueError):
        wg = as_weighted(g,
                         edge_weight,
                         cache_weights=True,
                         write_weights_through=True)
Ejemplo n.º 2
0
def test_anyhashableg_as_weighted():
    g = create_graph(
        directed=False,
        allowing_self_loops=True,
        allowing_multiple_edges=False,
        weighted=False,
        any_hashable=True,
    )

    g.add_vertex('0')
    g.add_vertex('1')
    g.add_edge('0', '1', edge='0')

    with pytest.raises(ValueError):
        g.set_edge_weight('0', 5.0)

    assert g.get_edge_weight('0') == 1.0

    def edge_weight(e):
        return 100.5

    wg = as_weighted(g,
                     edge_weight,
                     cache_weights=False,
                     write_weights_through=False)

    assert wg.get_edge_weight('0') == 100.5

    print(g)
    print(wg)

    with pytest.raises(ValueError):
        wg.set_edge_weight('0', 5.0)
Ejemplo n.º 3
0
def test_anyhashableg_as_weighted_with_None_function():
    g = create_graph(
        directed=False,
        allowing_self_loops=True,
        allowing_multiple_edges=False,
        weighted=False,
        any_hashable=True,
    )

    g.add_vertex(0)
    g.add_vertex(1)
    g.add_edge(0, 1, edge=0)

    with pytest.raises(ValueError):
        g.set_edge_weight(0, 5.0)

    assert g.get_edge_weight(0) == 1.0

    def edge_weight(e):
        return 100.5

    wg = as_weighted(g,
                     edge_weight_cb=None,
                     cache_weights=False,
                     write_weights_through=False)

    assert wg.get_edge_weight(0) == 1.0

    with pytest.raises(ValueError):
        wg.set_edge_weight(0, 5.0)
Ejemplo n.º 4
0
def test_as_weighted_with_no_caching_and_write_through():
    g = create_graph(
        directed=False,
        allowing_self_loops=True,
        allowing_multiple_edges=False,
        weighted=True,
    )

    g.add_vertex(0)
    g.add_vertex(1)
    g.add_edge(0, 1)

    g.set_edge_weight(0, 5.0)
    assert g.get_edge_weight(0) == 5.0

    def edge_weight(e):
        return 100.5

    wg = as_weighted(g,
                     edge_weight,
                     cache_weights=False,
                     write_weights_through=True)

    assert wg.get_edge_weight(0) == 100.5

    with pytest.raises(ValueError):
        wg.set_edge_weight(0, 5.0)

    assert wg.get_edge_weight(0) == 100.5