Ejemplo n.º 1
0
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]}
Ejemplo n.º 2
0
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]}
Ejemplo n.º 3
0
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]}