From 5058655a326e868f58713da441618aab99aa84dd Mon Sep 17 00:00:00 2001
From: Matt Wala <wala1@illinois.edu>
Date: Fri, 16 Mar 2018 15:13:44 -0500
Subject: [PATCH] Fix a misleading assertion in
test_pickling_with_module_import(). Only globals in the module are pickled
---
test/test_py_codegen.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/test/test_py_codegen.py b/test/test_py_codegen.py
index b418cc4..0015e7a 100644
--- a/test/test_py_codegen.py
+++ b/test/test_py_codegen.py
@@ -8,14 +8,13 @@ import sys
def test_pickling_with_module_import():
cg = codegen.PythonCodeGenerator()
- cg("import pytools.py_codegen")
+ cg("import pytools")
cg("import math as m")
import pickle
mod = pickle.loads(pickle.dumps(cg.get_picklable_module()))
assert mod.mod_globals["pytools"] is pytools
- assert mod.mod_globals["pytools"].py_codegen is pytools.py_codegen
import math
assert mod.mod_globals["m"] is math
--
GitLab