test rng in parallel

pathos/tests/test_random.py

@@ -0,0 +1,63 @@
+#!/usr/bin/env python
+#
+# Author: Mike McKerns (mmckerns @caltech and @uqfoundation)
+# Copyright (c) 2023 The Uncertainty Quantification Foundation.
+# License: 3-clause BSD.  The full license text is available at:
+#  - https://github.com/uqfoundation/pathos/blob/master/LICENSE
+
+from pathos.pools import *
+
+def random(x):
+    from pathos.helpers import mp_helper as mp
+    return mp.random_state('random').random()+x
+
+def rand(x):
+    from pathos.helpers import mp_helper as mp
+    return mp.random_state('numpy.random').rand()+x
+
+def wrong1(x):
+    import random
+    return random.random()+x
+
+def wrong2(x):
+    import numpy
+    return numpy.random.rand()+x
+
+
+def check_random(pool):
+    res = pool.map(random, range(2))
+    assert res[0] != res[1]
+    res = pool.map(rand, range(2))
+    assert res[0] != res[1]
+    pool.close()
+    pool.join()
+    pool.clear()
+    return
+
+
+def check_wrong(pool):
+    res = pool.map(wrong1, range(2))
+    assert res[0] == res[1]
+    res = pool.map(wrong2, range(2))
+    assert res[0] == res[1]
+    pool.close()
+    pool.join()
+    pool.clear()
+    return
+
+
+def test_random():
+    check_random(ThreadPool())
+    check_random(ProcessPool())
+    check_random(ParallelPool())
+
+def test_wrong():
+    check_random(ProcessPool())
+    check_random(ParallelPool())
+
+
+if __name__ == '__main__':
+    from pathos.helpers import freeze_support, shutdown
+    freeze_support()
+    test_random()
+    shutdown()