def test_5(self):
w = aobj.change_shape(self.ca, (ds.DiagAxis("f", 3),) + ds.CPLX)
self.assertAllclose(w, np.array([[[1, -10], [10, 1]],
[[2, -20], [20, 2]],
[[3, -30], [30, 3]]]))
self.assertEqual(w.dims, (ds.DiagAxis("f", 3),) + ds.CPLX)
def test_3(self):
dims = (ds.DiagAxis("p", 2), ds.DiagAxis("f", 3), )
w = aobj.change_shape(self.b, dims)
self.assertAllclose(w, self.b.T)
self.assertEqual(w.dims, self.b.dims[::-1])
def test_4(self):
w = aobj.change_shape(self.a, (ds.DiagAxis("f", 3),) + ds.CPLX)
self.assertAllclose(w, np.array([[[1, 0], [0, 0]],
[[2, 0], [0, 0]],
[[3, 0], [0, 0]]]))
self.assertEqual(w.dims, (ds.DiagAxis("f", 3),) + ds.CPLX)
def test_1(self):
w = aobj.change_shape(self.a, (ds.DiagAxis("f", 3),))
self.assertAllclose(w, self.a)
self.assertEqual(w.dims, self.a.dims)
