misc fixes

inducer · Jun 25, 2024 · 4f6a814 · 4f6a814
1 parent 0052e21
commit 4f6a814
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/meshmode/distributed.py b/meshmode/distributed.py
@@ -415,9 +415,11 @@ def membership_list_to_map(
     stored as a sorted :class:`numpy.ndarray`).
     """
     from pytools import unique
+
+    # FIXME: not clear why the sorted() call is necessary here
     return {
         entry: np.where(membership_list == entry)[0]
-        for entry in unique(list(membership_list))}
+        for entry in sorted(unique(membership_list))}
 
 
 # FIXME: Move somewhere else, since it's not strictly limited to distributed?

diff --git a/test/test_partition.py b/test/test_partition.py
@@ -455,7 +455,7 @@ def _test_connected_parts(mpi_comm, connected_parts):
     for i_remote_part in range(num_parts):
         if all_connected_masks[i_remote_part][mpi_comm.rank]:
             parts_connected_to_me.add(i_remote_part)
-    assert parts_connected_to_me == connected_parts
+    assert parts_connected_to_me == set(connected_parts)
 
 
 # TODO