xonsh/tests/test_imphooks.py
Bob Hyman 26475146e2
Flake8 1 (#3550)
* flake8 fixes -- tests only

* fix ci failure

* integrate fix from is_3551 so tests will pass.

* Update tests/test_builtins.py

Co-authored-by: Gil Forsyth <gforsyth@users.noreply.github.com>
2020-05-05 06:42:28 -04:00

72 lines
1.5 KiB
Python

# -*- coding: utf-8 -*-
"""Testing xonsh import hooks"""
import os
import builtins
from importlib import import_module
import pytest
from xonsh import imphooks
from xonsh.execer import Execer
from xonsh.environ import Env
from xonsh.built_ins import unload_builtins
imphooks.install_import_hooks()
@pytest.fixture(autouse=True)
def imp_env(xonsh_builtins):
Execer(unload=False)
builtins.__xonsh__.env = Env({"PATH": [], "PATHEXT": []})
yield
unload_builtins()
def test_import():
import sample
assert "hello mom jawaka\n" == sample.x
def test_absolute_import():
from xpack import sample
assert "hello mom jawaka\n" == sample.x
def test_relative_import():
from xpack import relimp
assert "hello mom jawaka\n" == relimp.sample.x
assert "hello mom jawaka\ndark chest of wonders" == relimp.y
def test_sub_import():
from xpack.sub import sample
assert "hello mom jawaka\n" == sample.x
TEST_DIR = os.path.dirname(__file__)
def test_module_dunder_file_attribute():
import sample
exp = os.path.join(TEST_DIR, "sample.xsh")
assert os.path.abspath(sample.__file__) == exp
def test_module_dunder_file_attribute_sub():
from xpack.sub import sample
exp = os.path.join(TEST_DIR, "xpack", "sub", "sample.xsh")
assert os.path.abspath(sample.__file__) == exp
def test_get_source():
mod = import_module('sample')
loader = mod.__loader__
source = loader.get_source('sample')
with open(os.path.join(TEST_DIR, 'sample.xsh'), 'rt') as srcfile:
assert source == srcfile.read()