67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
# Copyright 2017 the authors.
|
|
# This file is part of Hy, which is free software licensed under the Expat
|
|
# license. See the LICENSE.
|
|
|
|
from hy.importer import (import_file_to_module, import_buffer_to_ast,
|
|
MetaLoader, get_bytecode_path)
|
|
from hy.errors import HyTypeError
|
|
import os
|
|
import ast
|
|
import tempfile
|
|
|
|
|
|
def test_basics():
|
|
"Make sure the basics of the importer work"
|
|
import_file_to_module("basic",
|
|
"tests/resources/importer/basic.hy")
|
|
|
|
|
|
def test_stringer():
|
|
_ast = import_buffer_to_ast("(defn square [x] (* x x))", '')
|
|
assert type(_ast.body[0]) == ast.FunctionDef
|
|
|
|
|
|
def test_imports():
|
|
path = os.getcwd() + "/tests/resources/importer/a.hy"
|
|
testLoader = MetaLoader(path)
|
|
|
|
def _import_test():
|
|
try:
|
|
return testLoader.load_module("tests.resources.importer.a")
|
|
except:
|
|
return "Error"
|
|
|
|
assert _import_test() == "Error"
|
|
assert _import_test() is not None
|
|
|
|
|
|
def test_import_error_reporting():
|
|
"Make sure that (import) reports errors correctly."
|
|
|
|
def _import_error_test():
|
|
try:
|
|
import_buffer_to_ast("(import \"sys\")", '')
|
|
except HyTypeError:
|
|
return "Error reported"
|
|
|
|
assert _import_error_test() == "Error reported"
|
|
assert _import_error_test() is not None
|
|
|
|
|
|
def test_import_autocompiles():
|
|
"Test that (import) byte-compiles the module."
|
|
|
|
f = tempfile.NamedTemporaryFile(suffix='.hy', delete=False)
|
|
f.write(b'(defn pyctest [s] (+ "X" s "Y"))')
|
|
f.close()
|
|
|
|
try:
|
|
os.remove(get_bytecode_path(f.name))
|
|
except (IOError, OSError):
|
|
pass
|
|
import_file_to_module("mymodule", f.name)
|
|
assert os.path.exists(get_bytecode_path(f.name))
|
|
|
|
os.remove(f.name)
|
|
os.remove(get_bytecode_path(f.name))
|