Exemple #1
0
    def test_init(self):
        # empty list
        a = TensorCollection([])
        assert len(a) == 0

        # numpy array
        a = TensorCollection(np.ones((1, 2, 3)))
        assert len(a) == 1
        assert a.size == 2

        # nested list of numbers
        a = TensorCollection([[1, 2], [3, 4]])
        assert len(a) == 2
        assert a.size == 2

        # nested tuple of numbers
        a = TensorCollection(((1, 2), (3, 4)))
        assert len(a) == 2
        assert a.size == 2

        # nested list of Tensor objects
        a = TensorCollection([[Tensor(1, 2, 3), Tensor(3, 4, 5)]])
        assert a.shape == (1, 2, 3)
        assert len(a) == 1
        assert a.size == 2

        # object with __array__ function
        class A:
            def __array__(self):
                return np.array([Tensor(1, 2), Tensor(3, 4)])

        a = TensorCollection(A())
        assert len(a) == 2
        assert a.size == 2
Exemple #2
0
 def test_add_edge(self):
     a = Tensor([1, 0, 0, 0])
     b = Tensor([[42, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]],
                covariant=False)
     diagram = TensorDiagram((a, b))
     assert diagram.calculate() == Tensor([42, 0, 0, 0])
     diagram.add_edge(a.copy(), b)
     assert diagram.calculate() == 42
Exemple #3
0
    def test_getitem(self):
        a = Tensor([[1, 2], [3, 4]])
        b = Tensor([[5, 6], [7, 8]])
        c = TensorCollection([a, b])

        assert c[0] == a
        assert c[1] == b
        assert list(c) == [a, b]
        assert c[:, 1] == TensorCollection([Tensor([3, 4]), Tensor([7, 8])])
        assert c[:, 0, 0] == [1, 5]
Exemple #4
0
    def test_tensor_product(self):
        e1 = Tensor(1, 0)
        e2 = Tensor(0, 1)
        a = Tensor([0, 1], [1, 0], covariant=[0])
        b = Tensor([1, 0], [0, 1], covariant=[0])

        m = a.tensor_product(b)
        e = e1.tensor_product(e2)
        assert TensorDiagram(
            (e, m), (e, m)).calculate() == (a * e1).tensor_product(b * e2)

        d = TensorDiagram()
        d.add_node(a)
        d.add_node(b)
        assert d.calculate() == a.tensor_product(b)
Exemple #5
0
    def test_getitem(self):
        a = Tensor([[1, 2], [3, 4]], covariant=[0])

        assert a[0, 1] == 2
        assert a[None, 1] == [[3, 4]]
        assert a[None, 1].tensor_shape == (0, 1)
        assert a[::-1, 0] == [3, 1]
        assert a[::-1, 0].tensor_shape == (1, 0)
Exemple #6
0
    def test_arithmetic(self):
        a = Tensor(2, 3)
        b = Tensor(5, 4)

        # vector operations
        assert a + b == Tensor(7, 7)
        assert a - b == Tensor(-3, -1)
        assert -a == Tensor(-2, -3)

        # scalar operations
        assert a + 6 == Tensor(8, 9)
        assert a - 6 == Tensor(-4, -3)
        assert a * 6 == Tensor(12, 18)
        assert a / 6 == Tensor(1 / 3, 0.5)
Exemple #7
0
    def test_flat(self):
        a = [Tensor([[1, 2], [3, 4]]), Tensor([[5, 6], [7, 8]])]
        b = TensorCollection([a], tensor_rank=2)

        assert list(b.flat) == a
Exemple #8
0
 def __array__(self):
     return np.array([Tensor(1, 2), Tensor(3, 4)])
Exemple #9
0
    def test_dtype(self):
        a = Tensor(2, 3, dtype=np.float32)
        assert a.dtype == np.float32

        a = Tensor(2, 3, dtype=np.complex64)
        assert a.dtype == np.complex64
Exemple #10
0
    def test_transpose(self):
        a = Tensor([[1, 2], [3, 4]], covariant=[0])

        assert a.transpose() == Tensor([[1, 3], [2, 4]])
        assert a.T._covariant_indices == {1}
        assert a.T.T == a