adjust test_maps to run with coverage

uqfoundation · Sep 5, 2022 · a30244b · a30244b
1 parent 5214e9a
commit a30244b
Showing 1 changed file with 12 additions and 10 deletions.
diff --git a/pathos/tests/test_maps.py b/pathos/tests/test_maps.py
@@ -69,14 +69,16 @@ def added(*x):
 assert dill.copy(s)(squared, range(4)) == dill.copy(p)(squared, range(4)).get()
 del s, p
 
-from pathos.pools import _ProcessPool as _Pool
-s = Smap(Pool)
-p = Map(_Pool)
-r = s(p, [[squared, range(4)]]*4)
-del s, p
+import os
+if not os.environ.get('COVERAGE'): #XXX: travis-ci
+    from pathos.pools import _ProcessPool as _Pool
+    s = Smap(Pool)
+    p = Map(_Pool)
+    r = s(p, [[squared, range(4)]]*4)
+    del s, p
 
-s = Amap(Pool)
-p = Imap(_Pool)
-t = s(lambda x: list(p(squared, x)), [range(4)]*4)
-assert r == t.get()
-del s, p, r, t
+    s = Amap(Pool)
+    p = Imap(_Pool)
+    t = s(lambda x: list(p(squared, x)), [range(4)]*4)
+    assert r == t.get()
+    del s, p, r, t