Merge pull request #901 from EvelynMS1/fix814err

fix #814

Author: mwcraig

AUTHORS.rst

@@ -46,6 +46,7 @@ Alphabetical list of code contributors
 * Pey Lian Lim (@pllim)
 * James McCormac (@jmccormac01)
 * Abigale Moen (@AbigaleMoen)
+* Evelyn Montes Sobal (@EvelynMS1)
 * Stefan Nelson (@stefannelson)
 * Alex Niemi (@AlexN1234)
 * Joe Philip Ninan (@indiajoe)

ccdproc/image_collection.py

@@ -432,8 +432,12 @@ def filter(self, **kwd):
             A new collection with the files matched by the arguments
             to filter.
         """
+
         files = self.files_filtered(include_path=True, **kwd)
-        return ImageFileCollection(filenames=files, keywords=self.keywords)
+
+        return ImageFileCollection(
+            filenames=files, keywords=self.keywords, ext=self._ext
+        )
 
     def _get_files(self):
         """Helper method which checks whether ``files`` should be set

ccdproc/tests/test_image_collection.py

@@ -1152,3 +1152,17 @@ def test_filter_on_regex_escape_characters(self, triage_setup):
             new_ic = ic.filter(regex_fl=kw)
             assert len(new_ic.files) == 1
             assert kw in new_ic.summary["regex_fl"]
+
+    def test_filter_on_check_ext(self, tmp_path):
+        """
+        Testing the 814 error
+        """
+        hdul = fits.HDUList(
+            [fits.PrimaryHDU(np.ones((10, 10))), fits.ImageHDU(np.ones((10, 10)))]
+        )
+        hdul.writeto(tmp_path / "mef.fits")
+        ifc = ImageFileCollection(tmp_path, ext=1)
+        assert len(ifc.summary) > 0
+        ifc2 = ifc.filter()
+        assert len(ifc2.summary) > 0
+        assert ifc2.ext == 1