Merge pull request #213 from astronomy-commons/sean/fix-soap-catalog-…

…info Fix soap arguments to catalog info not including all parameters
astronomy-commons · Feb 16, 2024 · dd63dfe · dd63dfe
2 parents d36a212 + 3b5a29c
commit dd63dfe
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
@@ -31,6 +31,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
@@ -58,3 +59,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