IO tests: add test_ProcessFitsImage

MPoL-dev · Nov 30, 2023 · 8a7340b · 8a7340b
1 parent 21ea3d5
commit 8a7340b
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/test/input_output_test.py b/test/input_output_test.py
@@ -4,3 +4,14 @@
 
 from mpol.input_output import ProcessFitsImage
 
+def test_ProcessFitsImage():
+    # get a .fits file produced with casa
+    fname = download_file(
+        "https://zenodo.org/record/4711811/files/logo_cube.tclean.fits",
+        cache=True,
+        show_progress=True,
+        pkgname="mpol",
+    )
+
+    fits_image = ProcessFitsImage(fname)
+    clean_im, clean_im_ext, clean_beam = fits_image.get_image(beam=True)