def test_get_dataset(self):
    MixtureRegistry.add('test_mix3', [(self.cached_task.name, 1)])
    task_ds = TaskRegistry.get_dataset(self.cached_task.name, {'inputs': 13, 'targets': 13}, 'validation', use_cached=False, shuffle=False)
    mix_ds = MixtureRegistry.get('test_mix3').get_dataset({'inputs': 13, 'targets': 13}, 'validation', use_cached=False, shuffle=False)
    task_ds = task_ds.map((lambda x: {k: x[k] for k in ['inputs', 'targets']}))
    test_utils.assert_datasets_eq(task_ds.repeat(2), mix_ds.take(4))
def test_get_dataset(self):
    MixtureRegistry.add('test_mix3', [(self.cached_task.name, 1)])
    task_ds = TaskRegistry.get_dataset(self.cached_task.name, {'inputs': 13, 'targets': 13}, 'validation', use_cached=False, shuffle=False)
    mix_ds = MixtureRegistry.get('test_mix3').get_dataset({'inputs': 13, 'targets': 13}, 'validation', use_cached=False, shuffle=False)
    task_ds = task_ds.map((lambda x: {k: x[k] for k in ['inputs', 'targets']}))
    test_utils.assert_datasets_eq(task_ds.repeat(2), mix_ds.take(4))