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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
import inspect
import sys
import textwrap
import tokenize
import traceback
import unittest2 as unittest
try:
from cStringIO import StringIO
except ImportError:
from StringIO import StringIO
from nose.inspector import inspect_traceback, Expander, tbsource
class TestExpander(unittest.TestCase):
def test_simple_inspect_frame(self):
src = StringIO('a > 2')
lc = { 'a': 2}
gb = {}
exp = Expander(lc, gb)
tokenize.tokenize(src.readline, exp)
# print "'%s'" % exp.expanded_source
self.assertEqual(exp.expanded_source.strip(), '2 > 2')
def test_inspect_traceback_continued(self):
a = 6
out = ''
try:
assert a < 1, \
"This is a multline expression"
except AssertionError:
et, ev, tb = sys.exc_info()
out = inspect_traceback(tb)
# print "'%s'" % out.strip()
self.assertEqual(out.strip(),
'>> assert 6 < 1, \\\n '
'"This is a multline expression"')
def test_get_tb_source_simple(self):
# no func frame
try:
assert False
except AssertionError:
et, ev, tb = sys.exc_info()
lines, lineno = tbsource(tb, 1)
self.assertEqual(''.join(lines).strip(), 'assert False')
self.assertEqual(lineno, 0)
def test_get_tb_source_func(self):
# func frame
def check_even(n):
print n
assert n % 2 == 0
try:
check_even(1)
except AssertionError:
et, ev, tb = sys.exc_info()
lines, lineno = tbsource(tb)
out = textwrap.dedent(''.join(lines))
self.assertEqual(out,
' print n\n'
' assert n % 2 == 0\n'
'try:\n'
' check_even(1)\n'
'except AssertionError:\n'
' et, ev, tb = sys.exc_info()\n'
)
self.assertEqual(lineno, 3)
# FIXME 2 func frames
def test_pick_tb_lines(self):
try:
val = "fred"
def defred(n):
return n.replace('fred','')
assert defred(val) == 'barney', "Fred - fred != barney?"
except AssertionError:
et, ev, tb = sys.exc_info()
out = inspect_traceback(tb)
# print "'%s'" % out.strip()
self.assertEqual(out.strip(),
">> assert defred('fred') == 'barney', "
'"Fred - fred != barney?"')
try:
val = "fred"
def defred(n):
return n.replace('fred','')
assert defred(val) == 'barney', \
"Fred - fred != barney?"
def refred(n):
return n + 'fred'
except AssertionError:
et, ev, tb = sys.exc_info()
out = inspect_traceback(tb)
#print "'%s'" % out.strip()
self.assertEqual(out.strip(),
">> assert defred('fred') == 'barney', "
'\\\n "Fred - fred != barney?"')
S = {'setup':1}
def check_even(n, nn):
assert S['setup']
print n, nn
assert n % 2 == 0 or nn % 2 == 0
try:
check_even(1, 3)
except AssertionError:
et, ev, tb = sys.exc_info()
out = inspect_traceback(tb)
print "'%s'" % out.strip()
self.assertEqual(out.strip(),
"assert {'setup': 1}['setup']\n"
" print 1, 3\n"
">> assert 1 % 2 == 0 or 3 % 2 == 0")
def test_bug_95(self):
"""Test that inspector can handle multi-line docstrings"""
try:
"""docstring line 1
docstring line 2
"""
a = 2
assert a == 4
except AssertionError:
et, ev, tb = sys.exc_info()
out = inspect_traceback(tb)
print "'%s'" % out.strip()
self.assertEqual(out.strip(),
"2 = 2\n"
">> assert 2 == 4")
if __name__ == '__main__':
#import logging
#logging.basicConfig()
#logging.getLogger('').setLevel(10)
unittest.main()
|
