summaryrefslogtreecommitdiffstats
path: root/Lib/test
diff options
context:
space:
mode:
authorMeador Inge <meadori@gmail.com>2011-11-26 05:36:48 (GMT)
committerMeador Inge <meadori@gmail.com>2011-11-26 05:36:48 (GMT)
commit6f16660ca7f813f47bcc2462a8517819b7b76f5c (patch)
tree16a7b303bfdb28a9a43fd9f114605fe3b13b817c /Lib/test
parent11e381310bb5f498186098855f843c2e07e7ea82 (diff)
downloadcpython-6f16660ca7f813f47bcc2462a8517819b7b76f5c.zip
cpython-6f16660ca7f813f47bcc2462a8517819b7b76f5c.tar.gz
cpython-6f16660ca7f813f47bcc2462a8517819b7b76f5c.tar.bz2
Issue #12618: create unit tests for the py_compile module
Diffstat (limited to 'Lib/test')
-rw-r--r--Lib/test/test_py_compile.py42
1 files changed, 42 insertions, 0 deletions
diff --git a/Lib/test/test_py_compile.py b/Lib/test/test_py_compile.py
new file mode 100644
index 0000000..7491587
--- /dev/null
+++ b/Lib/test/test_py_compile.py
@@ -0,0 +1,42 @@
+import imp
+import os
+import py_compile
+import shutil
+import tempfile
+import unittest
+
+from test import support, script_helper
+
+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'
+ self.cache_path = imp.cache_from_source(self.source_path)
+ 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))
+ self.assertFalse(os.path.exists(self.cache_path))
+
+ def test_cache_path(self):
+ py_compile.compile(self.source_path)
+ self.assertTrue(os.path.exists(self.cache_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))
+ self.assertFalse(os.path.exists(self.cache_path))
+
+def test_main():
+ support.run_unittest(PyCompileTests)
+
+if __name__ == "__main__":
+ test_main()