Add tests for pyc compilation

This commit is contained in:
Vladimir Gorbunov 2013-04-13 20:57:54 +04:00
parent 3e0ff4a9b2
commit 0b03733c65

View File

@ -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'