def METHOD_NAME(self, split, src, tgt, dev_set, test_set):
    with tempfile.TemporaryDirectory() as root_dir:
        expected_samples = _get_mock_dataset(os.path.join(root_dir, "datasets"), split, src, tgt, dev_set, test_set)
        dataset = IWSLT2016(
            root=root_dir,
            split=split,
            language_pair=(src, tgt),
            valid_set=dev_set,
            test_set=test_set,
        )
        samples = list(dataset)
        for sample, expected_sample in zip_equal(samples, expected_samples):
            self.assertEqual(sample, expected_sample)