Add test to read fhd_source_array.sav files

RadioAstronomySoftwareGroup · Jun 13, 2023 · 7aed6ae · 7aed6ae
1 parent e5f808f
commit 7aed6ae
Showing 1 changed file with 13 additions and 9 deletions.
diff --git a/pyradiosky/tests/test_skymodel.py b/pyradiosky/tests/test_skymodel.py
@@ -2689,16 +2689,20 @@ def test_read_votable_errors():
 
 
 def test_fhd_catalog_reader():
-    catfile = os.path.join(SKY_DATA_PATH, "fhd_catalog.sav")
-
-    with uvtest.check_warnings(
-        UserWarning, match="Source IDs are not unique. Defining unique IDs."
-    ):
-        skyobj = SkyModel.from_fhd_catalog(catfile, expand_extended=False)
+    for fname in ["catalog", "source_array"]:
+        catfile = os.path.join(SKY_DATA_PATH, f"fhd_{fname}.sav")
+
+        if fname == "catalog":
+            with uvtest.check_warnings(
+                UserWarning, match="Source IDs are not unique. Defining unique IDs."
+            ):
+                skyobj = SkyModel.from_fhd_catalog(catfile, expand_extended=False)
+        else:
+            skyobj = SkyModel.from_fhd_catalog(catfile, expand_extended=False)
 
-    assert skyobj.filename == ["fhd_catalog.sav"]
-    catalog = scipy.io.readsav(catfile)["catalog"]
-    assert skyobj.Ncomponents == len(catalog)
+        assert skyobj.filename == [f"fhd_{fname}.sav"]
+        catalog = scipy.io.readsav(catfile)[fname]
+        assert skyobj.Ncomponents == len(catalog)
 
 
 def test_fhd_catalog_reader_extended_sources():