Ejemplo n.º 1
0
def test_switch_backend(backend):
    a = tn.Node(np.random.rand(3, 3, 3), name="A", backend="numpy")
    b = tn.Node(np.random.rand(3, 3, 3), name="B", backend="numpy")
    c = tn.Node(np.random.rand(3, 3, 3), name="C", backend="numpy")
    nodes = [a, b, c]
    tn.switch_backend(nodes, backend)
    assert nodes[0].backend.name == backend
def test_switch_backend_raises_error(num_charges):
  np.random.seed(10)
  a = tn.Node(
      get_random((3, 3, 3), num_charges=num_charges, dtype=np.float64),
      backend='symmetric')
  with pytest.raises(NotImplementedError):
    tn.switch_backend({a}, 'numpy')
Ejemplo n.º 3
0
def test_switch_backend():
    np.random.seed(10)
    a = tn.Node(np.random.rand(3, 3, 3), name="A", backend="numpy")
    b = tn.Node(np.random.rand(3, 3, 3), name="B", backend="numpy")
    c = tn.Node(np.random.rand(3, 3, 3), name="C", backend="numpy")
    nodes = [a, b, c]
    with pytest.raises(ValueError):
        tn.switch_backend(nodes, 'symmetric')
Ejemplo n.º 4
0
def test_switch_backend_raises_error(backend):
    a = tn.Node(np.random.rand(3, 3, 3))
    a.backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        tn.switch_backend({a}, backend)
Ejemplo n.º 5
0
def test_switch_backend_raises_error_2():
    np.random.seed(10)
    a = tn.Node(np.random.rand(3, 3, 3))
    with pytest.raises(ValueError):
        tn.switch_backend({a}, 'symmetric')