Skip to content

Commit d49aa0e

Browse files
committed
Issue #128: Update action_desat test
1 parent 5e1305c commit d49aa0e

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

tests/integration/envs/general_satellite_tasking/scenario/test_int_sat_actions.py

+6-5
Original file line numberDiff line numberDiff line change
@@ -139,18 +139,19 @@ def make_env(self):
139139
data_manager=data.NoDataManager(),
140140
sim_rate=1.0,
141141
max_step_duration=300.0,
142-
time_limit=300.0,
142+
time_limit=1200.0,
143143
disable_env_checker=True,
144144
)
145145

146146
def test_desat_action(self):
147147
env = self.make_env()
148148
env.reset()
149149
init_speeds = env.satellite.dynamics.wheel_speeds
150-
env.step(0) # Desat
151-
assert np.linalg.norm(env.satellite.dynamics.wheel_speeds) < np.linalg.norm(
152-
init_speeds
153-
)
150+
for _ in range(4):
151+
env.step(0)
152+
current_speeds = env.satellite.dynamics.wheel_speeds
153+
assert np.linalg.norm(current_speeds) < np.linalg.norm(init_speeds)
154+
init_speeds = current_speeds
154155

155156
def test_desat_action_power_draw(self):
156157
env = self.make_env()

0 commit comments

Comments
 (0)