TST fix test for cloudpickle <= 3.7

cloudpipe · Mar 27, 2019 · 33bb384 · 33bb384
1 parent a34c11c
commit 33bb384
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/cloudpickle_test.py b/tests/cloudpickle_test.py
@@ -899,7 +899,11 @@ def f():
         # some setup is required to allow pytest apimodules to be correctly
         # serializable.
         from cloudpickle import CloudPickler
-        CloudPickler.dispatch[type(py.builtin)] = cloudpickle.module_reduce
+        if sys.version_info[:2] >= (3, 8):
+            CloudPickler.dispatch[type(py.builtin)] = cloudpickle.module_reduce
+        else:
+            CloudPickler.dispatch[type(py.builtin)] = CloudPickler.save_module
+
         g = cloudpickle.loads(cloudpickle.dumps(f, protocol=self.protocol))
 
         result = g()