xonsh/tests/test_imphooks.py
pre-commit-ci[bot] 66c0490d37
[pre-commit.ci] pre-commit autoupdate (#5271)
* [pre-commit.ci] pre-commit autoupdate

updates:
- [github.com/psf/black: 23.12.1 → 24.1.1](https://github.com/psf/black/compare/23.12.1...24.1.1)

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2024-01-30 12:23:50 +01:00

71 lines
1.4 KiB
Python

"""Testing xonsh import hooks"""
import os
from importlib import import_module
import pytest
from xonsh import imphooks
@pytest.fixture(autouse=True)
def imp_env(xession):
xession.env.update({"PATH": [], "PATHEXT": []})
imphooks.install_import_hooks(xession.execer)
yield
def test_import():
import sample
assert "hello mom jawaka\n" == sample.x
def test_import_empty():
from xpack import empty_xsh
assert empty_xsh
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")) as srcfile:
assert source == srcfile.read()