def test_2(self):
ca = aobj.make_fullcomplex_array(self.ca)
self.assertAllclose(ca, np.array([[[1, -10],
[10, 1]],
[[2, -20],
[20, 2]],
[[3, -30],
[30, 3]]]))
self.assertEqual(ca.dims, (ds.DiagAxis("f", 3),) + ds.CPLX)
def test_1(self):
a = aobj.make_fullcomplex_array(self.a)
self.assertAllclose(a, np.array([[[1, 0],
[0, 0]],
[[2, 0],
[0, 0]],
[[3, 0],
[0, 0]]]))
self.assertEqual(a.dims, (ds.DiagAxis("f", 3),) + ds.CPLX)
def test_4(self):
d = aobj.make_fullcomplex_array(np.array(1 + 1j))
self.assertAllclose(d, np.array([[1, -1], [1, 1]]))
self.assertEqual(d.dims, ds.CPLX)
def test_3(self):
d = aobj.make_fullcomplex_array(self.d)
self.assertAllclose(d, self.d)
self.assertEqual(d.dims, self.d.dims)
