def setUp(self):
super().setUp()
# Input of the DownloadManager
self.dl_results = {}
self.dl_fnames = {}
self.extract_results = PathDict()
# Track calls to downloader/extractor
self.downloaded_urls = []
self.extracted_paths = []
# Virtual file system
self.fs = testing.MockFs() # Dict file_path -> file_content
# Start all mocks
self.fs.__enter__()
self._make_downloader_mock().start()
self._make_extractor_mock().start()
self.addCleanup(mock.patch.stopall)
def setUp(self):
super(DownloadManagerTest, self).setUp()
# Input of the DownloadManager
self.dl_results = {}
self.dl_fnames = {}
self.extract_results = {}
# Track calls to downloader/extractor
self.downloaded_urls = []
self.extracted_paths = []
# Virtual file system
self.fs = testing.MockFs() # Dict file_path -> file_content
# Start all mocks
self.fs.mock().start()
absltest.mock.patch.object(checksums_lib, 'store_checksums').start()
self._make_downloader_mock().start()
self._make_extractor_mock().start()
self.addCleanup(absltest.mock.patch.stopall)
def setUp(self):
super().setUp()
# Input of the DownloadManager
self.dl_results = {}
self.dl_fnames = {}
self.extract_results = PathDict()
# Track calls to downloader/extractor
self.downloaded_urls = []
self.extracted_paths = []
# Virtual file system
self.fs = testing.MockFs() # Dict file_path -> file_content
# Start all mocks
self.fs.__enter__()
self._make_downloader_mock().start()
self._make_extractor_mock().start()
# Mock `_checksum_paths` (do not load the precomputed checksums)
mock.patch.object(checksums_lib, '_checksum_paths', lambda: {}).start()
self.addCleanup(mock.patch.stopall)
def mock_fs() -> Iterator[testing.MockFs]:
"""Patch `tf.io.gfile` API into a virtual file system."""
with testing.MockFs() as fs:
yield fs