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]}
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]}
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]}