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