Skip to content

Commit

Permalink
Fix unit test and remove print statement
Browse files Browse the repository at this point in the history
  • Loading branch information
IshaanDesai committed Feb 12, 2025
1 parent e386411 commit 123060f
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 13 deletions.
20 changes: 10 additions & 10 deletions micro_manager/micro_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -209,16 +209,16 @@ def solve(self) -> None:
self._global_ids_of_local_sims[inactive_id]
)

print(
"Rank {} has active sims: {}".format(
self._rank, active_sim_gids
)
)
print(
"Rank {} has inactive sims: {}".format(
self._rank, inactive_sim_gids
)
)
# print(
# "Rank {} has active sims: {}".format(
# self._rank, active_sim_gids
# )
# )
# print(
# "Rank {} has inactive sims: {}".format(
# self._rank, inactive_sim_gids
# )
# )

if self._is_adaptivity_with_load_balancing:
if n % self._load_balancing_n == 0 and not first_time_window:
Expand Down
4 changes: 1 addition & 3 deletions tests/unit/test_micro_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -105,9 +105,7 @@ 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

0 comments on commit 123060f

Please sign in to comment.