def test_headers_save_location(self, triage_setup):
collection = ImageFileCollection(location=triage_setup.test_dir,
keywords=['imagetyp'])
destination = mkdtemp()
for header in collection.headers(save_location=destination):
pass
new_collection = ImageFileCollection(location=destination,
keywords=['imagetyp'])
basenames = lambda paths: set(
[os.path.basename(file) for file in paths])
assert (len(basenames(collection._paths()) -
basenames(new_collection._paths())) == 0)
rmtree(destination)
def test_generator_headers_save_with_name(self, triage_setup):
collection = ImageFileCollection(location=triage_setup.test_dir,
keywords=['imagetyp'])
for header in collection.headers(save_with_name='_new'):
assert isinstance(header, fits.Header)
new_collection = ImageFileCollection(location=triage_setup.test_dir,
keywords=['imagetyp'])
assert (len(new_collection._paths()) ==
2 * (triage_setup.n_test['files']) -
triage_setup.n_test['compressed'])
[os.remove(fil) for fil in iglob(triage_setup.test_dir + '/*_new*')]
def test_generator_full_path(self, triage_setup):
collection = ImageFileCollection(
location=triage_setup.test_dir, keywords=['imagetyp'])
for path, file_name in zip(collection._paths(), collection.files):
assert path == os.path.join(triage_setup.test_dir, file_name)
