Skip to content

Commit

Permalink
Adding import_file tests
Browse files Browse the repository at this point in the history
  • Loading branch information
jsiirola committed Dec 13, 2019
1 parent 7a74660 commit 859df86
Show file tree
Hide file tree
Showing 4 changed files with 26 additions and 0 deletions.
3 changes: 3 additions & 0 deletions pyutilib/misc/tests/import_exception.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
import sys

raise RuntimeError("raised during import")
5 changes: 5 additions & 0 deletions pyutilib/misc/tests/import_main_exception.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
import sys

if __name__ == "__main__":
print("import_main_exception - main")
raise RuntimeError("raised from __main__")
1 change: 1 addition & 0 deletions pyutilib/misc/tests/import_main_exception.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
import_main_exception - main
17 changes: 17 additions & 0 deletions pyutilib/misc/tests/test_import.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,6 +54,18 @@ def test_run_file3(self):
currdir + "import2.txt")[0])
os.remove(currdir + "import2.log")

def test_run_file_exception(self):
with self.assertRaisesRegexp(RuntimeError, "raised from __main__"):
pyutilib.misc.run_file(
"import_main_exception.py",
logfile=currdir + "import_main_exception.log", execdir=currdir)

self.assertFalse(
pyutilib.misc.comparison.compare_file(
currdir + "import_main_exception.log",
currdir + "import_main_exception.txt")[0])
os.remove(currdir + "import_main_exception.log")


class TestImportFile(unittest.TestCase):

Expand Down Expand Up @@ -83,6 +95,11 @@ def test_import_file_context3(self):
if not "import1" in globals():
self.fail("test_import_file - failed to import the import1.py file")

def test_run_file_exception(self):
with self.assertRaisesRegexp(RuntimeError, "raised during import"):
pyutilib.misc.run_file(
"import_exception.py", execdir=currdir)

def test1(self):
try:
pyutilib.misc.import_file('tfile.py')
Expand Down

0 comments on commit 859df86

Please sign in to comment.