def test_download_all_chunks(self):
responses_add_dem_mocks()
ws_dir, tgt_path = self.prepare_single_file_download()
with open(tgt_path, 'wb') as tgt:
rd.download(tgt, ws_dir, expect_res=(20,10))
self.check_single_file_download(24, tgt_path, ws_dir)
def test_download_two_chunks(self):
self.responses_add_mocks()
ws_dir, tgt_path = self.prepare_single_file_download()
with open(tgt_path, 'wb') as tgt:
rd.download(tgt, ws_dir, chunks=('A', 'X'), expect_res=(8,8))
self.check_single_file_download(2, tgt_path, ws_dir)
def download_all_chunks_separately(self, prefix=None, expect_prefix='15-',
expect_suffix='.tif'):
ws_dir, chunk_dir = self.prepare_multiple_chunk_download()
# Kick off download
rd.download(None, ws_dir, expect_res=(20,10),
chunk_directory=chunk_dir, chunk_prefix=prefix)
self.check_multiple_chunk_download(24, ws_dir, chunk_dir, expect_prefix, expect_suffix)