25 lines
539 B
Python
25 lines
539 B
Python
|
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'
|