Skip to content

Commit

Permalink
fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
IshaanDesai committed Dec 19, 2024
1 parent b8dfc46 commit e222e9f
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 3 deletions.
4 changes: 3 additions & 1 deletion tests/unit/test_micro_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -105,7 +105,9 @@ def test_solve_micro_sims(self):
manager._micro_sims = [MicroSimulation(i) for i in range(4)]
manager._micro_sims_active_steps = np.zeros(4, dtype=np.int32)

micro_sims_output = manager._solve_micro_simulations(self.fake_read_data, 1.0)
micro_sims_output, _ = manager._solve_micro_simulations(
self.fake_read_data, 1.0
)

for data, fake_data in zip(micro_sims_output, self.fake_write_data):
self.assertEqual(data["micro-scalar-data"], 2)
Expand Down
4 changes: 2 additions & 2 deletions tests/unit/test_micro_simulation_crash_handling.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ def test_crash_handling(self):
)
manager._micro_sims = [MicroSimulation(i) for i in range(4)]

micro_sims_output = manager._solve_micro_simulations(macro_data, 1.0)
micro_sims_output, _ = manager._solve_micro_simulations(macro_data, 1.0)

# Crashed simulation has interpolated value
data_crashed = micro_sims_output[2]
Expand Down Expand Up @@ -106,7 +106,7 @@ def test_crash_handling_with_adaptivity(self):
[-2, -2, -2, -2, 2]
)

micro_sims_output = manager._solve_micro_simulations_with_adaptivity(
micro_sims_output, _ = manager._solve_micro_simulations_with_adaptivity(
macro_data, 1.0
)

Expand Down

0 comments on commit e222e9f

Please sign in to comment.