def test_repr(self): self.zeros = ConstantArray([10, 20, 40, 50]) self.a = TransposedArray(self.zeros, [1, 3, 0, 2]) expected = ("TransposedArray(<ConstantArray shape=(10, 20, 40, 50) " "dtype=dtype('float64')>, [1, 3, 0, 2])") self.assertEqual(repr(self.a), expected)
def setUp(self): self.orig_array = np.empty([5, 6, 4, 7]) self.arr_transposed = self.orig_array.transpose([1, 3, 0, 2]) self.a = TransposedArray(self.orig_array, [1, 3, 0, 2]) self.expected_shape = (6, 7, 5, 4)
def setUp(self): self.zeros = ConstantArray([10, 20, 40, 50]) self.a = TransposedArray(self.zeros, [1, 3, 0, 2])
def transposed_shape(self, shape, axes): return TransposedArray(ConstantArray(shape), axes).shape
def test_axes_non_list(self): arr = TransposedArray(self.orig_array, tuple(self.a.axes)) result = arr[:2, 0, :, np.newaxis, :] self.assertEqual(result.shape, (2, 5, 1, 4))