Issue #126: Fix broken tests

AVSLab · Jul 30, 2024 · b328ad5 · b328ad5
1 parent cfc26ef
commit b328ad5
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/tests/unittest/utils/test_orbital.py b/tests/unittest/utils/test_orbital.py
@@ -53,13 +53,13 @@ class TestWalkerDeltaArgs:
     def test_randomize_ta_and_lan(self):
         sats = [MagicMock(), MagicMock()]
         walker_delta_arg_setup = orbital.walker_delta_args(
-            randomize_true_anomaly=True, randomize_lan=True
+            n_planes=1, randomize_true_anomaly=True, randomize_lan=True
         )
         sat_arg_map = walker_delta_arg_setup(sats)
-        assert sat_arg_map[sats[0]].f == sat_arg_map[sats[1]].f
-        assert sat_arg_map[sats[0]].f != 0
-        assert sat_arg_map[sats[0]].Omega == sat_arg_map[sats[1]].Omega
-        assert sat_arg_map[sats[0]].Omega != 0
+        assert sat_arg_map[sats[0]]["oe"].f == sat_arg_map[sats[1]]["oe"].f
+        assert sat_arg_map[sats[0]]["oe"].f != 0
+        assert sat_arg_map[sats[0]]["oe"].Omega == sat_arg_map[sats[1]]["oe"].Omega
+        assert sat_arg_map[sats[0]]["oe"].Omega != 0
 
     @patch(
         "bsk_rl.utils.orbital.walker_delta",
@@ -68,13 +68,13 @@ def test_randomize_ta_and_lan(self):
     def test_dont_randomize_ta_and_lan(self):
         sats = [MagicMock(), MagicMock()]
         walker_delta_arg_setup = orbital.walker_delta_args(
-            randomize_true_anomaly=False, randomize_lan=False
+            n_planes=1, randomize_true_anomaly=False, randomize_lan=False
         )
         sat_arg_map = walker_delta_arg_setup(sats)
-        assert sat_arg_map[sats[0]].f == sat_arg_map[sats[1]].f
-        assert sat_arg_map[sats[0]].f == 0
-        assert sat_arg_map[sats[0]].Omega == sat_arg_map[sats[1]].Omega
-        assert sat_arg_map[sats[0]].Omega == 0
+        assert sat_arg_map[sats[0]]["oe"].f == sat_arg_map[sats[1]]["oe"].f
+        assert sat_arg_map[sats[0]]["oe"].f == 0
+        assert sat_arg_map[sats[0]]["oe"].Omega == sat_arg_map[sats[1]]["oe"].Omega
+        assert sat_arg_map[sats[0]]["oe"].Omega == 0
 
 
 class TestRandomEpoch: