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