fix soap arguments to catalog info not including all parameters

astronomy-commons · Feb 10, 2024 · 3b5a29c · 3b5a29c
1 parent 5b25b88
commit 3b5a29c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/hipscat_import/soap/arguments.py b/src/hipscat_import/soap/arguments.py
@@ -61,6 +61,7 @@ def to_catalog_info(self, total_rows) -> AssociationCatalogInfo:
             "primary_catalog": str(self.object_catalog_dir),
             "join_column": self.source_object_id_column,
             "join_catalog": str(self.source_catalog_dir),
+            "contains_leaf_files": self.write_leaf_files,
         }
         return AssociationCatalogInfo(**info)
 

diff --git a/tests/hipscat_import/soap/test_run_soap.py b/tests/hipscat_import/soap/test_run_soap.py
@@ -30,6 +30,7 @@ def test_object_to_source(dask_client, small_sky_soap_args):
     assert catalog.catalog_path == small_sky_soap_args.catalog_path
     assert len(catalog.get_join_pixels()) == 14
     assert catalog.catalog_info.total_rows == 17161
+    assert not catalog.catalog_info.contains_leaf_files
 
 
 @pytest.mark.dask
@@ -57,3 +58,4 @@ def test_object_to_source_with_leaves(
     assert catalog.catalog_path == small_sky_soap_args.catalog_path
     assert len(catalog.get_join_pixels()) == 14
     assert catalog.catalog_info.total_rows == 17161
+    assert catalog.catalog_info.contains_leaf_files