def test_cached_data_source_shuffle_buffer_default(self):
    self._mock_and_assert_cached_source('cached_task', None)
def test_cached_data_source_shuffle_buffer_default(self):
    self._mock_and_assert_cached_source('cached_task', None)