diff --git a/src/corrgi/alignment.py b/src/corrgi/alignment.py index 4c4213b..ea8667b 100644 --- a/src/corrgi/alignment.py +++ b/src/corrgi/alignment.py @@ -35,7 +35,8 @@ def autocorrelation_alignment(catalog: Catalog) -> PixelAlignment: [ [pix.order, pix.pixel, pix.order, pix.pixel, pix.order, pix.pixel] for pix in catalog.get_healpix_pixels() - ] + ], + columns=column_names, ) result_mapping = pd.concat([upper_triangle, diagonal]) return PixelAlignment(catalog.pixel_tree, result_mapping, PixelAlignmentType.OUTER) diff --git a/tests/corrgi/test_alignment.py b/tests/corrgi/test_alignment.py index 8840b44..e4ff359 100644 --- a/tests/corrgi/test_alignment.py +++ b/tests/corrgi/test_alignment.py @@ -6,6 +6,7 @@ def test_autocorrelation_alignment(data_catalog_dir): data_catalog = hipscat.read_from_hipscat(data_catalog_dir) alignment = autocorrelation_alignment(data_catalog) assert len(alignment.pixel_mapping) == 28 + assert len(alignment.pixel_mapping.columns) == 6 def test_crosscorrelation_alignment(dr7_lrg_catalog_dir, dr7_lrg_rand_catalog_dir): @@ -16,3 +17,4 @@ def test_crosscorrelation_alignment(dr7_lrg_catalog_dir, dr7_lrg_rand_catalog_di ## dr7_rand_catalog has 22 partitions ## 12*21 = 252 assert len(alignment.pixel_mapping) == 252 + assert len(alignment.pixel_mapping.columns) == 6