def test_5(self): a = aobj.remove_tail(np.array(self.d)) self.assertAllclose(a, 0) self.assertEqual(a.shape, (2, 12 * 25))
def test_2(self): a = aobj.remove_tail(self.c) self.assertAllclose(a, 0) self.assertEqual(a.dims, (ds.DimSweep("f", 2), ds.DimRep("Tail", 12)))
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)
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)))