add 2D transpose tests

Author: v923z

tests/2d/numpy/transpose.py

@@ -0,0 +1,58 @@
+try:
+    from ulab import numpy as np
+except ImportError:
+    import numpy as np
+
+dtypes = (np.uint8, np.int8, np.uint16, np.int16, np.float)
+
+print("Basic 2D transpose")
+a = np.array(range(6)).reshape((2, 3))
+print("Original array:")
+print(a)
+print("Transposed (no axes):")
+print(a.transpose())
+print()
+
+print("2D transpose with axes=(1, 0)")
+a = np.array(range(6)).reshape((2, 3))
+print("Original array:")
+print(a)
+print("Transposed with axes=(1, 0):")
+print(a.transpose(axes=(1, 0)))
+print()
+
+print("2D transpose with axes=(0, 1)")
+a = np.array(range(6)).reshape((2, 3))
+print("Original array:")
+print(a)
+print("Transposed with axes=(0, 1):")
+print(a.transpose(axes=(0, 1)))
+print()
+
+print("1D array transpose")
+a = np.array(range(5))
+print("Original array:")
+print(a)
+print("Transposed array:")
+print(a.transpose())
+print()
+
+print("Transpose of different dtypes")
+for dtype in dtypes:
+    a = np.array(range(4), dtype=dtype).reshape((2, 2))
+    print('Original array:')
+    print(a)
+    print('Transposed array:')
+    print(a.transpose())
+print()
+
+print(".T property vs transpose() method")
+a = np.array(range(6)).reshape((2, 3))
+print("Original array:")
+print(a)
+print("Using .T property:")
+print(a.T)
+print("Using .transpose():")
+print(a.transpose())
+print("Both should be equal:", np.all(a.T == a.transpose()))
+print()

tests/2d/numpy/transpose.py.exp

@@ -0,0 +1,78 @@
+Basic 2D transpose
+Original array:
+array([[0.0, 1.0, 2.0],
+       [3.0, 4.0, 5.0]], dtype=float64)
+Transposed (no axes):
+array([[0.0, 3.0],
+       [1.0, 4.0],
+       [2.0, 5.0]], dtype=float64)
+
+2D transpose with axes=(1, 0)
+Original array:
+array([[0.0, 1.0, 2.0],
+       [3.0, 4.0, 5.0]], dtype=float64)
+Transposed with axes=(1, 0):
+array([[0.0, 3.0],
+       [1.0, 4.0],
+       [2.0, 5.0]], dtype=float64)
+
+2D transpose with axes=(0, 1)
+Original array:
+array([[0.0, 1.0, 2.0],
+       [3.0, 4.0, 5.0]], dtype=float64)
+Transposed with axes=(0, 1):
+array([[0.0, 1.0, 2.0],
+       [3.0, 4.0, 5.0]], dtype=float64)
+
+1D array transpose
+Original array:
+array([0.0, 1.0, 2.0, 3.0, 4.0], dtype=float64)
+Transposed array:
+array([0.0, 1.0, 2.0, 3.0, 4.0], dtype=float64)
+
+Transpose of different dtypes
+Original array:
+array([[0, 1],
+       [2, 3]], dtype=uint8)
+Transposed array:
+array([[0, 2],
+       [1, 3]], dtype=uint8)
+Original array:
+array([[0, 1],
+       [2, 3]], dtype=int8)
+Transposed array:
+array([[0, 2],
+       [1, 3]], dtype=int8)
+Original array:
+array([[0, 1],
+       [2, 3]], dtype=uint16)
+Transposed array:
+array([[0, 2],
+       [1, 3]], dtype=uint16)
+Original array:
+array([[0, 1],
+       [2, 3]], dtype=int16)
+Transposed array:
+array([[0, 2],
+       [1, 3]], dtype=int16)
+Original array:
+array([[0.0, 1.0],
+       [2.0, 3.0]], dtype=float64)
+Transposed array:
+array([[0.0, 2.0],
+       [1.0, 3.0]], dtype=float64)
+
+.T property vs transpose() method
+Original array:
+array([[0.0, 1.0, 2.0],
+       [3.0, 4.0, 5.0]], dtype=float64)
+Using .T property:
+array([[0.0, 3.0],
+       [1.0, 4.0],
+       [2.0, 5.0]], dtype=float64)
+Using .transpose():
+array([[0.0, 3.0],
+       [1.0, 4.0],
+       [2.0, 5.0]], dtype=float64)
+Both should be equal: True
+