tests: fix a test for session saving (#492)

uqfoundation · Jun 7, 2022 · 260b372 · 260b372
1 parent 74de347
commit 260b372
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/__main__.py b/tests/__main__.py
@@ -26,5 +26,5 @@
     for test in tests:
         p = sp.Popen([python, test], shell=shell).wait()
         if not p:
-            print('.', end='')
+            print('.', end='', flush=True)
     print('')
diff --git a/tests/test_session.py b/tests/test_session.py
@@ -209,14 +209,14 @@ def test_objects(main, copy_dict, byref):
         dump = test_file.getvalue()
         test_file.close()
 
-        sys.modules[modname] = ModuleType(modname)  # empty
+        main = sys.modules[modname] = ModuleType(modname)  # empty
         # This should work after fixing https://github.com/uqfoundation/dill/issues/462
         test_file = dill._dill.StringIO(dump)
-        dill.load_session(test_file)
+        dill.load_session(test_file, main=main)
     finally:
         test_file.close()
 
-    assert x == 42
+    assert main.x == 42
 
 
     # Dump session for module that is not __main__: