From 89749407af49c5f4dc41910608e65f4e7bbe4d87 Mon Sep 17 00:00:00 2001 From: Meador Inge Date: Fri, 25 Nov 2011 23:28:18 -0600 Subject: [PATCH] Issue #12618: create unit tests for the py_compile module --- Lib/test/test_py_compile.py | 43 +++++++++++++++++++++++++++++++++++++ 1 file changed, 43 insertions(+) create mode 100644 Lib/test/test_py_compile.py diff --git a/Lib/test/test_py_compile.py b/Lib/test/test_py_compile.py new file mode 100644 index 0000000000..6e6dc487b9 --- /dev/null +++ b/Lib/test/test_py_compile.py @@ -0,0 +1,43 @@ +import imp +import os +import py_compile +import shutil +import tempfile +import unittest + +from test import test_support + +class PyCompileTests(unittest.TestCase): + + def setUp(self): + self.directory = tempfile.mkdtemp() + self.source_path = os.path.join(self.directory, '_test.py') + self.pyc_path = self.source_path + 'c' + with open(self.source_path, 'w') as file: + file.write('x = 123\n') + + def tearDown(self): + shutil.rmtree(self.directory) + + def test_absolute_path(self): + py_compile.compile(self.source_path, self.pyc_path) + self.assertTrue(os.path.exists(self.pyc_path)) + + def test_cwd(self): + cwd = os.getcwd() + os.chdir(self.directory) + py_compile.compile(os.path.basename(self.source_path), + os.path.basename(self.pyc_path)) + os.chdir(cwd) + self.assertTrue(os.path.exists(self.pyc_path)) + + def test_relative_path(self): + py_compile.compile(os.path.relpath(self.source_path), + os.path.relpath(self.pyc_path)) + self.assertTrue(os.path.exists(self.pyc_path)) + +def test_main(): + test_support.run_unittest(PyCompileTests) + +if __name__ == "__main__": + test_main() -- 2.50.1