Add test to read fhd_source_array.sav files

RadioAstronomySoftwareGroup · Dec 14, 2023 · ef6a407 · ef6a407
1 parent 79e6900
commit ef6a407
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/tests/test_skymodel.py b/tests/test_skymodel.py
@@ -2945,16 +2945,20 @@ def test_read_votable_errors():
 
 
 def test_fhd_catalog_reader():
-    catfile = os.path.join(SKY_DATA_PATH, "fhd_catalog.sav")
+    for fname in ["catalog", "source_array"]:
+        catfile = os.path.join(SKY_DATA_PATH, f"fhd_{fname}.sav")
 
-    with uvtest.check_warnings(
-        UserWarning, match="Source IDs are not unique. Defining unique IDs."
-    ):
-        skyobj = SkyModel.from_fhd_catalog(catfile, expand_extended=False)
+        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)
 
 
 @pytest.mark.parametrize("extended", [True, False])