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_1(self): a = random_value_array(3, 5) m = aobj.make_vector(np.array(a), a.dims[:-1]) self.assertAllclose(m, a) self.assertEqual(m.dims[-1:], (ds.DimMatrix_j("j", a.shape[-1]),))
def test_1(self): a = random_value_array(3, 5) m = aobj.make_matrix(np.array(a), a.dims[:-2]) self.assertAllclose(m, a) self.assertEqual(m.dims[-2:], (ds.DimMatrix_i("i", a.shape[-2]), ds.DimMatrix_j("j", a.shape[-1])))
def test_4(self): d = aobj.remove_rep(self.d) self.assertAllclose(d, 0) dims = (ds.DimSweep("f", 2), ds.DimRep("AllReps", 12), ds.DimMatrix_i("i", 5), ds.DimMatrix_j("j", 5),) self.assertEqual(d.dims, dims)
def test_3(self): a = aobj.remove_tail(self.d) self.assertAllclose(a, 0) dims = (ds.DimSweep("f", 2), ds.DimRep("Tail", 12), ds.DimMatrix_i("i", 5), ds.DimMatrix_j("j", 5),) self.assertEqual(a.dims, dims)