From c3087eedf7d6668b55e81c1ad3cfd61d2aec196e Mon Sep 17 00:00:00 2001 From: Kyle Oman Date: Thu, 10 Oct 2024 20:24:30 +0100 Subject: [PATCH] Cover more catalogue attributes in soap ordering test. --- tests/test_iterator.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/tests/test_iterator.py b/tests/test_iterator.py index 8772944..8f7ea06 100644 --- 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, + )