Python get_parallel_edges Exemples

Exemple #1

Fichier : network_test.py Projet : zyzhang1992/TensorNetwork

def test_get_parallel_edge(backend):
  a = tn.Node(np.ones((2,) * 5), backend=backend)
  b = tn.Node(np.ones((2,) * 5), backend=backend)
  edges = set()
  for i in {0, 1, 3}:
    edges.add(tn.connect(a[i], b[i]))
  for e in edges:
    assert set(tn.get_parallel_edges(e)) == edges

Exemple #2

def test_get_parallel_edge(backend):
    a = tn.Node(np.ones((2, ) * 5), backend=backend)
    b = tn.Node(np.ones((2, ) * 5), backend=backend)
    edges = set()
    for i in {0, 1, 3}:
        edges.add(tn.connect(a[i], b[i]))
    # sort by edge signature
    e = sorted(list(edges))[0]
    assert tn.get_parallel_edges(e) == edges