From 0b03733c65e7284c0ca875d1f086ca60e6909231 Mon Sep 17 00:00:00 2001 From: Vladimir Gorbunov Date: Sat, 13 Apr 2013 20:57:54 +0400 Subject: [PATCH] Add tests for pyc compilation --- tests/importer/test_pyc.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 tests/importer/test_pyc.py diff --git a/tests/importer/test_pyc.py b/tests/importer/test_pyc.py new file mode 100644 index 0000000..85059ce --- /dev/null +++ b/tests/importer/test_pyc.py @@ -0,0 +1,24 @@ +import os +import sys +import imp +import tempfile +from hy.importer import write_hy_as_pyc + + +def test_pyc(): + """Test pyc compilation.""" + f = tempfile.NamedTemporaryFile(suffix='.hy', delete=False) + if sys.version_info[0] >= 3: + f.write(b'(defn pyctest [s] s)') + else: + f.write('(defn pyctest [s] s)') + f.close() + + write_hy_as_pyc(f.name) + os.unlink(f.name) + + cfile = "%s.pyc" % f.name[:-len(".hy")] + mod = imp.load_compiled('pyc', cfile) + os.unlink(cfile) + + assert mod.pyctest('Foo') == 'Foo'