Class Dataset_

tests/data/test_zipdataset.py:22–34

class Dataset_(torch.utils.data.Dataset):

    def __init__(self, length, index_only=True):
        self.len = length
        self.index_only = index_only

    def __len__(self):
        return self.len

    def __getitem__(self, index):
        if self.index_only:
            return index
        return 1, 2, index


TEST_CASE_1 = [[Dataset_(5), Dataset_(5), Dataset_(5)], None, (0, 0, 0), 5]