1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
from pytest import raises
from pypy.tool.pytest.astrewriter import ast_rewrite
def get_assert_explanation(space, src):
fn = "?"
w_code = ast_rewrite.rewrite_asserts(space, src, fn)
w_d = space.newdict(module=True)
excinfo = space.raises_w(space.w_AssertionError, space.exec_, w_code, w_d, w_d)
return space.text_w(space.getitem(space.getattr(excinfo.value.get_w_value(space), space.newtext("args")), space.newint(0)))
def test_simple(space):
src = """
x = 1
def f():
y = 2
assert x == y
f()
"""
expl = get_assert_explanation(space, src)
assert expl == 'assert 1 == 2'
def test_call(space):
src = """
x = 1
def g():
return 15
def f():
y = 2
assert g() == x + y
f()
"""
expl = get_assert_explanation(space, src)
assert expl == 'assert 15 == (1 == 2)\n + where 15 = g()'
def test_list(space):
src = """
x = 1
y = 2
assert [1, 1, x] == [1, 1, y]
"""
expl = get_assert_explanation(space, src)
# diff etc disabled for now
assert expl == 'assert [1, 1, 1] == [1, 1, 2]'
def test_boolop(space):
src = "x = 1; y = 2; assert x == 1 and y == 3"
expl = get_assert_explanation(space, src)
assert expl == 'assert (1 == 1 and 2 == 3)'
src = "x = 1; y = 2; assert x == 2 and y == 3"
expl = get_assert_explanation(space, src)
assert expl == 'assert (1 == 2)'
def test_attribute(space):
src = """
class A:
x = 1
def f():
a = A
assert a.x == 2
f()
"""
expl = get_assert_explanation(space, src)
assert expl == "assert 1 == 2\n + where 1 = <class 'A'>.x"
|
