Cover more catalogue attributes in soap ordering test.

SWIFTSIM · Oct 10, 2024 · c3087ee · c3087ee
1 parent 8b22391
commit c3087ee
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/tests/test_iterator.py b/tests/test_iterator.py
@@ -452,8 +452,20 @@ def test_soap_target_order_consistency(
             preload={"black_holes.masses"},  # just keep warnings quiet
         )
         for i, sg in enumerate(sgs):
-            sg_individual = sgs_individual[sgs.iteration_order[i]]
+            sg_single = sgs_individual[sgs.iteration_order[i]]
             assert_allclose_units(
-                sg_individual.halo_catalogue._region_centre,
+                sg_single.halo_catalogue._region_centre,
                 sg.halo_catalogue._region_centre,
             )
+            assert_allclose_units(
+                sg_single.halo_catalogue.centre,
+                sg.halo_catalogue.centre,
+            )
+            assert_allclose_units(
+                sg_single.halo_catalogue.velocity_centre,
+                sg.halo_catalogue.velocity_centre,
+            )
+            assert_allclose_units(
+                sg_single.halo_catalogue.spherical_overdensity_200_crit.centre_of_mass,
+                sg.halo_catalogue.spherical_overdensity_200_crit.centre_of_mass,
+            )