def testSqueeze(self): a1 = numpy.array([[[3,4,5]]]) a2 = a1.squeeze() a1 = ndarray_ext.squeeze(a1) self.assertEqual(a1.shape,a2.shape)