Adding tensor.__radd__. Making formatting of data in tensors of difference sizes consistent. Closes #18.

Author: dmdunla

TensorToolbox/tensor.py

@@ -265,8 +265,6 @@ def end(self, k=None):
         ----------
         k: int
             dimension for subscripted indexing
-        n: int
-            dimensions to index
 
         Returns
         -------
@@ -1380,6 +1378,21 @@ def tensor_add(x, y):
 
         return ttb.tt_tenfun(tensor_add, self, other)
 
+    def __radd__(self, other):
+        """
+        Reverse binary addition (+) for tensors
+
+        Parameters
+        ----------
+        other: :class:`TensorToolbox.tensor`, float, int
+
+        Returns
+        -------
+        :class:`TensorToolbox.tensor`
+        """
+
+        return self.__add__(other)
+
     def __pow__(self, power):
         """
         Element Wise Power (**) for tensors
@@ -1522,15 +1535,15 @@ def __repr__(self):
                 s += '[:]'
                 s += ' = \n'
                 s += str(self.data)
-                # s += '\n'
+                s += '\n'
                 return s
         for i, j in enumerate(range(0, np.prod(self.shape), self.shape[-1]*self.shape[-2])):
             s += 'data'
             if self.ndims == 2:
                 s += '[:, :]'
                 s += ' = \n'
                 s += str(self.data)
-                #s += '\n'
+                s += '\n'
             elif self.ndims > 2:
                 idx = ttb.tt_ind2sub(self.shape[:-2], np.array([i]))
                 s += str(idx[0].tolist())[0:-1]

tests/test_tensor.py

@@ -1074,7 +1074,7 @@ def test_tensor__str__(sample_tensor_2way):
     s += 'data'
     s += '[:] = \n'
     s += data.__str__()
-    # s += '\n'
+    s += '\n'
     assert s == tensorInstance.__str__()
 
     # Test 2D
@@ -1087,7 +1087,7 @@ def test_tensor__str__(sample_tensor_2way):
     s += 'data'
     s += '[:, :] = \n'
     s += data.__str__()
-    # s += '\n'
+    s += '\n'
     assert s == tensorInstance.__str__()
 
     # Test 3D,shape in decreasing and increasing order