test_datasets.py 文件源码

def testListDataset(self):
        def identity(x): return x
        h = [0, 1, 2]
        d = dataset.ListDataset(elem_list=h, load=identity)
        self.assertEqual(len(d), 3)
        self.assertEqual(d[0], 0)

        t = torch.LongTensor([0, 1, 2])
        d = dataset.ListDataset(elem_list=t, load=identity)
        self.assertEqual(len(d), 3)
        self.assertEqual(d[0], 0)

        a = np.asarray([0, 1, 2])
        d = dataset.ListDataset(elem_list=a, load=identity)
        self.assertEqual(len(d), 3)
        self.assertEqual(d[0], 0)