def setUp(self):
self.a = aobj.hfarray([1, 2, 3], dims=(ds.DiagAxis("f", 3),))
bdims = (ds.DiagAxis("f", 3), ds.DiagAxis("p", 3),)
self.b = aobj.hfarray([[1, 2, 3],
[10, 20, 30],
[100, 200, 300]],
dims=bdims)
def setUp(self):
self.a = aobj.hfarray([1, 2, 3], dims=(ds.DiagAxis("f", 3),))
dims = (ds.DiagAxis("f", 3),)
self.ca = aobj.hfarray([1 + 10j, 2 + 20j, 3 + 30j], dims=dims)
dims = (ds.DiagAxis("f", 3), ds.DiagAxis("p", 2),)
self.b = aobj.hfarray([[1, 2],
[10, 20],
[100, 200]], dims=dims)
self.d = aobj.hfarray([[1, -2], [2, 1]], dims=ds.CPLX)
def test_5(self):
w = aobj.expand_diagonals(self.b, (ds.DiagAxis("p", 3),))
B = np.zeros((3, 3, 3))
for i in range(3):
for j in range(3):
B[i, j, j] = (j + 1) * 10 ** i
self.assertAllclose(w, B)
dims = (ds.DiagAxis("f", 3), ds.IndepAxis("p", 3),
ds.DerivAxis("p", 3), )
self.assertEqual(w.dims, dims)
def setUp(self):
self.a = aobj.hfarray([1, 2, 3], dims=(ds.DiagAxis("f", 3),))
dims = (ds.DiagAxis("f", 3), ds.DiagAxis("p", 3),)
self.b = aobj.hfarray([[1, 2, 3],
[10, 20, 30],
[100, 200, 300]], dims=dims)
dims = (ds.DimSweep("f", 2), ds.DimSweep("Vg", 3), ds.DimRep("rep", 4))
self.c = aobj.hfarray(np.zeros((2, 3, 4)), dims=dims)
dims = (ds.DimSweep("f", 2), ds.DimSweep("Vg", 3), ds.DimRep("rep", 4),
ds.DimMatrix_i("i", 5), ds.DimMatrix_j("j", 5),)
self.d = aobj.hfarray(np.zeros((2, 3, 4, 5, 5)), dims=dims)
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):
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_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_1(self):
w = aobj.change_shape(self.a, (ds.DiagAxis("f", 3),))
self.assertAllclose(w, self.a)
self.assertEqual(w.dims, self.a.dims)
def test_2(self):
w = aobj.expand_diagonals(self.a, (ds.DiagAxis("f", 3),))
self.assertAllclose(w, np.array([[1, 0, 0], [0, 2, 0], [0, 0, 3]]))
dims = (ds.IndepAxis("f", 3), ds.DerivAxis("f", 3),)
self.assertEqual(w.dims, dims)
def test_4(self):
a, b = aobj.make_same_dims(self.a, np.array(self.a))
self.assertEqual(a.dims, b.dims)
self.assertEqual(a.dims, (ds.DiagAxis("f", 3), ))
def test_3(self):
b, a = aobj.make_same_dims(self.b, self.a)
self.assertEqual(a.dims, b.dims)
self.assertEqual(a.dims, (ds.DiagAxis("f", 3), ds.DiagAxis("p", 3),))
def test_2(self):
a, b = aobj.make_same_dims(self.a, self.b)
self.assertEqual(a.dims, b.dims)
self.assertEqual(a.dims, (ds.DiagAxis("f", 3), ds.DiagAxis("p", 3),))
def test_error_2(self):
self.assertRaises(Exception, aobj.expand_diagonals,
self.a, (ds.DiagAxis("p", 3),))
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_error_1(self):
dims = (ds.DiagAxis("f", 3), )
self.assertRaises(ValueError, aobj.change_shape, self.b, dims)
def test_1(self):
a = aobj.remove_tail(self.a)
self.assertAllclose(a, np.array(self.a)[:, newaxis])
self.assertEqual(a.dims, (ds.DiagAxis("f", 3), ds.DimRep("Tail", 1)))
