Skip to content

Commit 59eb0e7

Browse files
committed
adapt test to new format
1 parent a27f877 commit 59eb0e7

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

tests/unit/test_adaptivity_parallel.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -255,12 +255,11 @@ def test_get_ranks_of_sims(self):
255255
"""
256256
self._configurator.adaptivity_similarity_measure = MagicMock(return_value="L1")
257257

258+
expected_ranks_of_sims = {0: 0, 1: 0, 2: 0, 3: 1, 4: 1}
258259
if self._mpi.rank == 0:
259260
global_ids = [0, 1, 2]
260-
expected_ranks_of_sims = [0, 0, 0, 1, 1]
261261
elif self._mpi.rank == 1:
262262
global_ids = [3, 4]
263-
expected_ranks_of_sims = [0, 0, 0, 1, 1]
264263

265264
container = SimulationContainer(self._mpi)
266265
container.initialize(
@@ -271,4 +270,4 @@ def test_get_ranks_of_sims(self):
271270
)
272271

273272
actual_ranks_of_sims = container.get_ranks_of_sims()
274-
self.assertTrue(np.array_equal(expected_ranks_of_sims, actual_ranks_of_sims))
273+
self.assertDictEqual(expected_ranks_of_sims, actual_ranks_of_sims)

0 commit comments

Comments
 (0)