def METHOD_NAME(self):
    config = {
        'batch_size': 2,
        'drop_last': False,
        'shuffle': False,
        'num_workers': 4,
    }
    collate_fn = Collate_fn(config)
    ds = ListDataset()
    # test batch_size
    loader = Dataloader(
        ds,
        batch_size=config['batch_size'],
        drop_last=config['drop_last'],
        num_workers=config['num_workers'],
        collate_fn=collate_fn)
    epochs = 1
    for e in range(epochs):
        res = []
        for batch_data in loader:
            res.extend(batch_data['data'])
        self.assertEqual([i for i in range(DATA_SIZE)], res)