Python get_subgraph_dangling Exemples

Exemple #1

def test_subgraph_disconnected_nodes(backend):
    a = tn.Node(np.eye(2), backend=backend)
    b = tn.Node(np.eye(2), backend=backend)
    c = tn.Node(np.eye(2), backend=backend)
    a[0] ^ b[0]
    b[1] ^ c[1]
    edges = tn.get_subgraph_dangling({a, c})
    assert edges == {a[0], a[1], c[0], c[1]}

Exemple #2

def test_full_graph_subgraph_dangling(backend):
    a = tn.Node(np.eye(2), backend=backend)
    b = tn.Node(np.eye(2), backend=backend)
    c = tn.Node(np.eye(2), backend=backend)
    a[0] ^ b[0]
    b[1] ^ c[1]
    edges = tn.get_subgraph_dangling({a, b, c})
    assert edges == {a[1], c[0]}

Exemple #3

def test_subgraph_sanity(backend):
    a = tn.Node(np.eye(2), backend=backend)
    b = tn.Node(np.eye(2), backend=backend)
    a[0] ^ b[0]
    edges = tn.get_subgraph_dangling({a})
    assert edges == {a[0], a[1]}