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)