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)))