def test_files_with_no_compressed(self, triage_setup):
     collection = ImageFileCollection(location=triage_setup.test_dir)
     n_files_found = len(
         collection._fits_files_in_directory(compressed=False))
     n_uncompressed = (triage_setup.n_test['files'] -
                       triage_setup.n_test['compressed'])
     assert n_files_found == n_uncompressed
 def test_files_with_compressed(self, triage_setup):
     collection = ImageFileCollection(location=triage_setup.test_dir)
     assert len(collection._fits_files_in_directory(
         compressed=True)) == triage_setup.n_test['files']