fix: bad header init (#171)

lgrcia · Mar 25, 2024 · 4732901 · 4732901
2 parents eedfa9f + a5b669a
commit 4732901
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 5 deletions.
diff --git a/prose/core/image.py b/prose/core/image.py
@@ -71,11 +71,10 @@ def __post_init__(self):
         if self.computed is None:
             self.computed = {}
 
-        # backward compatibility
-        self.header = self.computed.get("fits_header", None)
-
         if self.header is None:
-            self.header = Header()
+            # backward compatibility
+            self.header = self.computed.get("fits_header", Header())
+
         if isinstance(self._sources, dict):
             self._sources = Sources(**self._sources)
         if self._sources is None:

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prose"
-version = "3.3.1"
+version = "3.3.2"
 description = "Modular image processing pipelines for Astronomy"
 authors = ["Lionel Garcia"]
 license = "MIT"

diff --git a/tests/test_image.py b/tests/test_image.py
@@ -54,3 +54,11 @@ def test_fitsimage(tmp_path):
 
     loaded_image = FITSImage(filename)
     assert "IMAGETYP" in dict(loaded_image.header)
+
+
+def test_init_header():
+
+    d = np.ones((512, 512))
+    hdr = dict(FILTER="B")
+
+    assert Image(d, header=hdr).header == hdr