340 lines
8.5 KiB
Python
340 lines
8.5 KiB
Python
"PYTEST_DONT_REWRITE"
|
|
import pytest, py
|
|
|
|
from _pytest.assertion import util
|
|
|
|
def exvalue():
|
|
return py.std.sys.exc_info()[1]
|
|
|
|
def f():
|
|
return 2
|
|
|
|
def test_not_being_rewritten():
|
|
assert "@py_builtins" not in globals()
|
|
|
|
def test_assert():
|
|
try:
|
|
assert f() == 3
|
|
except AssertionError:
|
|
e = exvalue()
|
|
s = str(e)
|
|
assert s.startswith('assert 2 == 3\n')
|
|
|
|
def test_assert_with_explicit_message():
|
|
try:
|
|
assert f() == 3, "hello"
|
|
except AssertionError:
|
|
e = exvalue()
|
|
assert e.msg == 'hello'
|
|
|
|
def test_assert_within_finally():
|
|
excinfo = py.test.raises(ZeroDivisionError, """
|
|
try:
|
|
1/0
|
|
finally:
|
|
i = 42
|
|
""")
|
|
s = excinfo.exconly()
|
|
assert py.std.re.search("division.+by zero", s) is not None
|
|
|
|
#def g():
|
|
# A.f()
|
|
#excinfo = getexcinfo(TypeError, g)
|
|
#msg = getmsg(excinfo)
|
|
#assert msg.find("must be called with A") != -1
|
|
|
|
|
|
def test_assert_multiline_1():
|
|
try:
|
|
assert (f() ==
|
|
3)
|
|
except AssertionError:
|
|
e = exvalue()
|
|
s = str(e)
|
|
assert s.startswith('assert 2 == 3\n')
|
|
|
|
def test_assert_multiline_2():
|
|
try:
|
|
assert (f() == (4,
|
|
3)[-1])
|
|
except AssertionError:
|
|
e = exvalue()
|
|
s = str(e)
|
|
assert s.startswith('assert 2 ==')
|
|
|
|
def test_in():
|
|
try:
|
|
assert "hi" in [1, 2]
|
|
except AssertionError:
|
|
e = exvalue()
|
|
s = str(e)
|
|
assert s.startswith("assert 'hi' in")
|
|
|
|
def test_is():
|
|
try:
|
|
assert 1 is 2
|
|
except AssertionError:
|
|
e = exvalue()
|
|
s = str(e)
|
|
assert s.startswith("assert 1 is 2")
|
|
|
|
|
|
@py.test.mark.skipif("sys.version_info < (2,6)")
|
|
def test_attrib():
|
|
class Foo(object):
|
|
b = 1
|
|
i = Foo()
|
|
try:
|
|
assert i.b == 2
|
|
except AssertionError:
|
|
e = exvalue()
|
|
s = str(e)
|
|
assert s.startswith("assert 1 == 2")
|
|
|
|
@py.test.mark.skipif("sys.version_info < (2,6)")
|
|
def test_attrib_inst():
|
|
class Foo(object):
|
|
b = 1
|
|
try:
|
|
assert Foo().b == 2
|
|
except AssertionError:
|
|
e = exvalue()
|
|
s = str(e)
|
|
assert s.startswith("assert 1 == 2")
|
|
|
|
def test_len():
|
|
l = list(range(42))
|
|
try:
|
|
assert len(l) == 100
|
|
except AssertionError:
|
|
e = exvalue()
|
|
s = str(e)
|
|
assert s.startswith("assert 42 == 100")
|
|
assert "where 42 = len([" in s
|
|
|
|
def test_assert_non_string_message():
|
|
class A:
|
|
def __str__(self):
|
|
return "hello"
|
|
try:
|
|
assert 0 == 1, A()
|
|
except AssertionError:
|
|
e = exvalue()
|
|
assert e.msg == "hello"
|
|
|
|
def test_assert_keyword_arg():
|
|
def f(x=3):
|
|
return False
|
|
try:
|
|
assert f(x=5)
|
|
except AssertionError:
|
|
e = exvalue()
|
|
assert "x=5" in e.msg
|
|
|
|
# These tests should both fail, but should fail nicely...
|
|
class WeirdRepr:
|
|
def __repr__(self):
|
|
return '<WeirdRepr\nsecond line>'
|
|
|
|
def bug_test_assert_repr():
|
|
v = WeirdRepr()
|
|
try:
|
|
assert v == 1
|
|
except AssertionError:
|
|
e = exvalue()
|
|
assert e.msg.find('WeirdRepr') != -1
|
|
assert e.msg.find('second line') != -1
|
|
assert 0
|
|
|
|
def test_assert_non_string():
|
|
try:
|
|
assert 0, ['list']
|
|
except AssertionError:
|
|
e = exvalue()
|
|
assert e.msg.find("list") != -1
|
|
|
|
def test_assert_implicit_multiline():
|
|
try:
|
|
x = [1,2,3]
|
|
assert x != [1,
|
|
2, 3]
|
|
except AssertionError:
|
|
e = exvalue()
|
|
assert e.msg.find('assert [1, 2, 3] !=') != -1
|
|
|
|
|
|
def test_assert_with_brokenrepr_arg():
|
|
class BrokenRepr:
|
|
def __repr__(self): 0 / 0
|
|
e = AssertionError(BrokenRepr())
|
|
if e.msg.find("broken __repr__") == -1:
|
|
py.test.fail("broken __repr__ not handle correctly")
|
|
|
|
def test_multiple_statements_per_line():
|
|
try:
|
|
a = 1; assert a == 2
|
|
except AssertionError:
|
|
e = exvalue()
|
|
assert "assert 1 == 2" in e.msg
|
|
|
|
def test_power():
|
|
try:
|
|
assert 2**3 == 7
|
|
except AssertionError:
|
|
e = exvalue()
|
|
assert "assert (2 ** 3) == 7" in e.msg
|
|
|
|
|
|
class TestView:
|
|
|
|
def setup_class(cls):
|
|
cls.View = pytest.importorskip("_pytest.assertion.oldinterpret").View
|
|
|
|
def test_class_dispatch(self):
|
|
### Use a custom class hierarchy with existing instances
|
|
|
|
class Picklable(self.View):
|
|
pass
|
|
|
|
class Simple(Picklable):
|
|
__view__ = object
|
|
def pickle(self):
|
|
return repr(self.__obj__)
|
|
|
|
class Seq(Picklable):
|
|
__view__ = list, tuple, dict
|
|
def pickle(self):
|
|
return ';'.join(
|
|
[Picklable(item).pickle() for item in self.__obj__])
|
|
|
|
class Dict(Seq):
|
|
__view__ = dict
|
|
def pickle(self):
|
|
return Seq.pickle(self) + '!' + Seq(self.values()).pickle()
|
|
|
|
assert Picklable(123).pickle() == '123'
|
|
assert Picklable([1,[2,3],4]).pickle() == '1;2;3;4'
|
|
assert Picklable({1:2}).pickle() == '1!2'
|
|
|
|
def test_viewtype_class_hierarchy(self):
|
|
# Use a custom class hierarchy based on attributes of existing instances
|
|
class Operation:
|
|
"Existing class that I don't want to change."
|
|
def __init__(self, opname, *args):
|
|
self.opname = opname
|
|
self.args = args
|
|
|
|
existing = [Operation('+', 4, 5),
|
|
Operation('getitem', '', 'join'),
|
|
Operation('setattr', 'x', 'y', 3),
|
|
Operation('-', 12, 1)]
|
|
|
|
class PyOp(self.View):
|
|
def __viewkey__(self):
|
|
return self.opname
|
|
def generate(self):
|
|
return '%s(%s)' % (self.opname, ', '.join(map(repr, self.args)))
|
|
|
|
class PyBinaryOp(PyOp):
|
|
__view__ = ('+', '-', '*', '/')
|
|
def generate(self):
|
|
return '%s %s %s' % (self.args[0], self.opname, self.args[1])
|
|
|
|
codelines = [PyOp(op).generate() for op in existing]
|
|
assert codelines == ["4 + 5", "getitem('', 'join')",
|
|
"setattr('x', 'y', 3)", "12 - 1"]
|
|
|
|
@py.test.mark.skipif("sys.version_info < (2,6)")
|
|
def test_assert_customizable_reprcompare(monkeypatch):
|
|
monkeypatch.setattr(util, '_reprcompare', lambda *args: 'hello')
|
|
try:
|
|
assert 3 == 4
|
|
except AssertionError:
|
|
e = exvalue()
|
|
s = str(e)
|
|
assert "hello" in s
|
|
|
|
def test_assert_long_source_1():
|
|
try:
|
|
assert len == [
|
|
(None, ['somet text', 'more text']),
|
|
]
|
|
except AssertionError:
|
|
e = exvalue()
|
|
s = str(e)
|
|
assert 're-run' not in s
|
|
assert 'somet text' in s
|
|
|
|
def test_assert_long_source_2():
|
|
try:
|
|
assert(len == [
|
|
(None, ['somet text', 'more text']),
|
|
])
|
|
except AssertionError:
|
|
e = exvalue()
|
|
s = str(e)
|
|
assert 're-run' not in s
|
|
assert 'somet text' in s
|
|
|
|
def test_assert_raise_alias(testdir):
|
|
testdir.makepyfile("""
|
|
"PYTEST_DONT_REWRITE"
|
|
import sys
|
|
EX = AssertionError
|
|
def test_hello():
|
|
raise EX("hello"
|
|
"multi"
|
|
"line")
|
|
""")
|
|
result = testdir.runpytest()
|
|
result.stdout.fnmatch_lines([
|
|
"*def test_hello*",
|
|
"*raise EX*",
|
|
"*1 failed*",
|
|
])
|
|
|
|
|
|
@pytest.mark.skipif("sys.version_info < (2,5)")
|
|
def test_assert_raise_subclass():
|
|
class SomeEx(AssertionError):
|
|
def __init__(self, *args):
|
|
super(SomeEx, self).__init__()
|
|
try:
|
|
raise SomeEx("hello")
|
|
except AssertionError:
|
|
s = str(exvalue())
|
|
assert 're-run' not in s
|
|
assert 'could not determine' in s
|
|
|
|
def test_assert_raises_in_nonzero_of_object_pytest_issue10():
|
|
class A(object):
|
|
def __nonzero__(self):
|
|
raise ValueError(42)
|
|
def __lt__(self, other):
|
|
return A()
|
|
def __repr__(self):
|
|
return "<MY42 object>"
|
|
def myany(x):
|
|
return True
|
|
try:
|
|
assert not(myany(A() < 0))
|
|
except AssertionError:
|
|
e = exvalue()
|
|
s = str(e)
|
|
assert "<MY42 object> < 0" in s
|
|
|
|
@py.test.mark.skipif("sys.version_info >= (2,6)")
|
|
def test_oldinterpret_importation():
|
|
# we had a cyclic import there
|
|
# requires pytest on sys.path
|
|
res = py.std.subprocess.call([
|
|
py.std.sys.executable, '-c', str(py.code.Source("""
|
|
try:
|
|
from _pytest.assertion.newinterpret import interpret
|
|
except ImportError:
|
|
from _pytest.assertion.oldinterpret import interpret
|
|
"""))
|
|
])
|
|
|
|
assert res == 0
|