update tests

LSSTDESC · Nov 19, 2024 · 92dfa4b · 92dfa4b
1 parent dcb153e
commit 92dfa4b
Showing 1 changed file with 24 additions and 24 deletions.
diff --git a/tests/test_dataops.py b/tests/test_dataops.py
@@ -914,12 +914,12 @@ def test_make_radial_profiles():
         )
         _test_profile_table_output(
             profile,
-            bins_radians[1],
-            expected_radius[1],
-            bins_radians[2],
-            expected["tan_shear"][1],
-            expected["cross_shear"][1],
-            [2],
+            bins_radians[:2],
+            expected_radius[:2],
+            bins_radians[1:3],
+            expected["tan_shear"][:2],
+            expected["cross_shear"][:2],
+            [1, 2],
         )
         # Test metadata
         assert_array_equal(profile.meta["bin_units"], bin_units)
@@ -935,12 +935,12 @@ def test_make_radial_profiles():
         )
         _test_profile_table_output(
             profile,
-            bins_radians[1],
-            expected_radius[1],
-            bins_radians[2],
-            expected["tan_shear"][1],
-            expected["cross_shear"][1],
-            [2],
+            bins_radians[:2],
+            expected_radius[:2],
+            bins_radians[1:3],
+            expected["tan_shear"][:2],
+            expected["cross_shear"][:2],
+            [1, 2],
         )
         # including empty bins
         profile = da.make_radial_profile(
@@ -995,12 +995,12 @@ def test_make_radial_profiles():
         # Test default behavior, remember that include_empty_bins=False excludes all bins with N>=1
         _test_profile_table_output(
             cluster.profile,
-            bins_radians[1],
-            expected_radius[1],
-            bins_radians[2],
-            expected["tan_shear"][1],
-            expected["cross_shear"][1],
-            [2],
+            bins_radians[:2],
+            expected_radius[:2],
+            bins_radians[1:3],
+            expected["tan_shear"][:2],
+            expected["cross_shear"][:2],
+            [1, 2],
             p0="gt",
             p1="gx",
         )
@@ -1073,12 +1073,12 @@ def test_make_radial_profiles():
         )
         _test_profile_table_output(
             cluster.profile3,
-            bins_radians[1],
-            expected_radius[1],
-            bins_radians[2],
-            expected["tan_shear"][1],
-            expected["cross_shear"][1],
-            [2],
+            bins_radians[:2],
+            expected_radius[:2],
+            bins_radians[1:3],
+            expected["tan_shear"][:2],
+            expected["cross_shear"][:2],
+            [1, 2],
             p0="gt",
             p1="gx",
         )