@@ -554,8 +554,8 @@ def test_polarization_is_invariant_to_translation(
554554 ):
555555 """Adding a constant offset does not change polarization."""
556556 shifted = partial_alignment_positions .copy ()
557- shifted .loc [dict ( space = "x" ) ] = shifted .sel (space = "x" ) + 1000.0
558- shifted .loc [dict ( space = "y" ) ] = shifted .sel (space = "y" ) - 500.0
557+ shifted .loc [{ " space" : "x" } ] = shifted .sel (space = "x" ) + 1000.0
558+ shifted .loc [{ " space" : "y" } ] = shifted .sel (space = "y" ) - 500.0
559559
560560 pol_original = kinematics .compute_polarization (
561561 partial_alignment_positions
@@ -597,8 +597,8 @@ def test_polarization_is_invariant_to_global_rotation(
597597 y = partial_alignment_positions .sel (space = "y" )
598598
599599 rotated = partial_alignment_positions .copy ()
600- rotated .loc [dict ( space = "x" ) ] = - y
601- rotated .loc [dict ( space = "y" ) ] = x
600+ rotated .loc [{ " space" : "x" } ] = - y
601+ rotated .loc [{ " space" : "y" } ] = x
602602
603603 pol_original = kinematics .compute_polarization (
604604 partial_alignment_positions
@@ -660,8 +660,8 @@ def test_body_axis_invariance_to_translation_scaling_rotation(
660660
661661 # Global translation: should not affect body-axis vectors.
662662 translated = da .copy ()
663- translated .loc [dict ( space = "x" ) ] = translated .sel (space = "x" ) + 123.4
664- translated .loc [dict ( space = "y" ) ] = translated .sel (space = "y" ) - 56.7
663+ translated .loc [{ " space" : "x" } ] = translated .sel (space = "x" ) + 123.4
664+ translated .loc [{ " space" : "y" } ] = translated .sel (space = "y" ) - 56.7
665665
666666 pol_translated , angle_translated = kinematics .compute_polarization (
667667 translated ,
@@ -698,8 +698,8 @@ def test_body_axis_invariance_to_translation_scaling_rotation(
698698 rotated = da .copy ()
699699 x = da .sel (space = "x" )
700700 y = da .sel (space = "y" )
701- rotated .loc [dict ( space = "x" ) ] = - y
702- rotated .loc [dict ( space = "y" ) ] = x
701+ rotated .loc [{ " space" : "x" } ] = - y
702+ rotated .loc [{ " space" : "y" } ] = x
703703
704704 pol_rotated , angle_rotated = kinematics .compute_polarization (
705705 rotated ,
@@ -1060,8 +1060,8 @@ def test_mean_angle_rotates_with_global_rotation(
10601060 y = partial_alignment_positions .sel (space = "y" )
10611061
10621062 rotated = partial_alignment_positions .copy ()
1063- rotated .loc [dict ( space = "x" ) ] = - y
1064- rotated .loc [dict ( space = "y" ) ] = x
1063+ rotated .loc [{ " space" : "x" } ] = - y
1064+ rotated .loc [{ " space" : "y" } ] = x
10651065
10661066 _ , angle_rotated = kinematics .compute_polarization (
10671067 rotated ,
0 commit comments