def test_getitem(self): var = Variable(value=np.random.rand(1, 1, 1, 1, 51), name='foo') slc = [slice(None, None, None), slice(None, None, None), slice(None, None, None), np.array([0]), np.array([14])] ret = var.__getitem__(slc) self.assertEqual(ret.shape, tuple([1] * 5))