def test_cached_data_source_shuffle_buffer_None(self):
    self.add_task('cached_task_buf_None', self.tfds_source, self._get_preps_with_cache_placeholder_buffer_size(None))
    shutil.copytree(self.cached_task_dir, os.path.join(self.test_data_dir, 'cached_task_buf_None'))
    self._mock_and_assert_cached_source('cached_task_buf_None', None)
def test_cached_data_source_shuffle_buffer_None(self):
    self.add_task('cached_task_buf_None', self.tfds_source, self._get_preps_with_cache_placeholder_buffer_size(None))
    shutil.copytree(self.cached_task_dir, os.path.join(self.test_data_dir, 'cached_task_buf_None'))
    self._mock_and_assert_cached_source('cached_task_buf_None', None)