100 lines
2.8 KiB
Python
100 lines
2.8 KiB
Python
|
import polyloader
|
||
|
|
import pytest
|
||
|
|
import py_compile
|
||
|
|
import ptutils
|
||
|
|
import stat
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
import imp
|
||
|
|
import random
|
||
|
|
import struct
|
||
|
|
|
||
|
|
class Compiler:
|
||
|
|
def __init__(self, pt):
|
||
|
|
self.pt = pt
|
||
|
|
|
||
|
|
def __call__(self, source_text, filename, *extra):
|
||
|
|
return compile("result='Success for %s: %s'" %
|
||
|
|
(self.pt, source_text.rstrip()), filename, "exec")
|
||
|
|
|
||
|
|
def __repr__(self):
|
||
|
|
return "Compiler %s" % (self.pt)
|
||
|
|
|
||
|
|
polyloader.install(Compiler("2"), ['2'])
|
||
|
|
|
||
|
|
TESTFN = '@test'
|
||
|
|
|
||
|
|
def clean_tmpfiles(path):
|
||
|
|
if os.path.exists(path):
|
||
|
|
os.remove(path)
|
||
|
|
if os.path.exists(path + 'c'):
|
||
|
|
os.remove(path + 'c')
|
||
|
|
if os.path.exists(path + 'o'):
|
||
|
|
os.remove(path + 'o')
|
||
|
|
|
||
|
|
def unload(name):
|
||
|
|
try:
|
||
|
|
del sys.modules[name]
|
||
|
|
except KeyError:
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class Test_RelativeImports:
|
||
|
|
|
||
|
|
def teardown_class(cls):
|
||
|
|
unload("tests.relimport")
|
||
|
|
|
||
|
|
def setup_class(cls):
|
||
|
|
unload("tests.relimport")
|
||
|
|
|
||
|
|
def test_relimport_star(self):
|
||
|
|
# This will import * from .test_import.
|
||
|
|
from . import relimport
|
||
|
|
assert(hasattr(relimport, "Test_Imports"))
|
||
|
|
|
||
|
|
def test_issue3221(self):
|
||
|
|
# Regression test for http://bugs.python.org/issue3221.
|
||
|
|
def check_absolute():
|
||
|
|
exec "from os import path" in ns
|
||
|
|
def check_relative():
|
||
|
|
exec "from . import relimport" in ns
|
||
|
|
|
||
|
|
# Check both OK with __package__ and __name__ correct
|
||
|
|
ns = dict(__package__='tests', __name__='test.notarealmodule')
|
||
|
|
check_absolute()
|
||
|
|
check_relative()
|
||
|
|
|
||
|
|
# Check both OK with only __name__ wrong
|
||
|
|
ns = dict(__package__='tests', __name__='notarealpkg.notarealmodule')
|
||
|
|
check_absolute()
|
||
|
|
check_relative()
|
||
|
|
|
||
|
|
# Check relative fails with only __package__ wrong
|
||
|
|
ns = dict(__package__='foo', __name__='test.notarealmodule')
|
||
|
|
with pytest.warns(RuntimeWarning) as rw:
|
||
|
|
check_absolute()
|
||
|
|
with pytest.raises(SystemError) as se:
|
||
|
|
check_relative()
|
||
|
|
|
||
|
|
# Check relative fails with __package__ and __name__ wrong
|
||
|
|
ns = dict(__package__='foo', __name__='notarealpkg.notarealmodule')
|
||
|
|
with pytest.warns(RuntimeWarning) as se:
|
||
|
|
check_absolute()
|
||
|
|
with pytest.raises(SystemError) as se:
|
||
|
|
check_relative()
|
||
|
|
|
||
|
|
# Check both fail with package set to a non-string
|
||
|
|
ns = dict(__package__=object())
|
||
|
|
with pytest.raises(ValueError) as ve:
|
||
|
|
check_absolute()
|
||
|
|
with pytest.raises(ValueError) as ve:
|
||
|
|
check_relative()
|
||
|
|
|
||
|
|
def test_absolute_import_without_future(self):
|
||
|
|
# If explicit relative import syntax is used, then do not try
|
||
|
|
# to perform an absolute import in the face of failure.
|
||
|
|
# Issue #7902.
|
||
|
|
with pytest.raises(ImportError) as ie:
|
||
|
|
from .os import sep
|
||
|
|
assert(False)
|