more robust _sym_tag_to_num_tag test

inducer · Sep 18, 2023 · e7f9239 · e7f9239
1 parent 6918c63
commit e7f9239
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/test/test_trace_pair.py b/test/test_trace_pair.py
@@ -87,7 +87,13 @@ class DerivedDerivedCommTag(DerivedCommTag):
     dct2 = DerivedCommTag()
     ddct = DerivedDerivedCommTag()
 
-    assert _sym_tag_to_num_tag(ct) == 441551355
+    try:
+        from mpi4py import MPI
+    except ModuleNotFoundError:
+        pass
+    else:
+        tag_ub = MPI.COMM_WORLD.Get_attr(MPI.TAG_UB)
+        assert _sym_tag_to_num_tag(ct) == (1549868734841116283675 % tag_ub)
 
     assert ct == ct2
     assert ct != dct