Skip to content

Commit f7b09e2

Browse files
committed
fix mada and lb test
1 parent be6ed4a commit f7b09e2

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

tests/unit/test_load_balancing.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -277,7 +277,7 @@ def test_redistribute_active_sims_two_ranks(self):
277277
load_balancer.balance()
278278

279279
actual_global_ids = [] #
280-
for lid in container.local_gids:
280+
for lid, gid in enumerate(container.local_gids):
281281
sim = container[lid]
282282
actual_global_ids.append(sim.get_global_id())
283283
self.assertListEqual(actual_global_ids, expected_global_ids)

tests/unit/test_model_adaptivity.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,7 @@ def switching_function(resolution, location, t, input, prev_output):
174174
def solve_variant(micro_sims_input, dt, computed_outputs):
175175
solve_calls.append(
176176
{
177-
"sim_name": manager._micro_sims[0].name,
177+
"sim_name": manager._sim_container[0].name,
178178
"computed_outputs": computed_outputs.copy(),
179179
}
180180
)

0 commit comments

Comments
 (0)