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
|
"""In this file, we test to ensure that the output of
check_syntax is as expected.
We also test to ensure that check_syntax does not accidently
change any existing error handling settings.
"""
import friendly_traceback as ft
def test_check_syntax():
# set-up
bad_code_syntax = "True = 1"
bad_code_exec = "a = b" # Not a syntax error, but a NameError
good_code = "c = 1"
ft.set_stream("capture")
original_include = ft.get_include()
installed = ft.is_installed()
# ----- end of set-up
# When a SyntaxError is raised, check_syntax returns False
assert not ft.editors_helpers.check_syntax(source=bad_code_syntax)
result = ft.get_output() # content is flushed
assert "SyntaxError" in result
assert not ft.get_output() # confirm that content was flushed
# When no SyntaxError is raised, check_syntax returns a tuple
# containing a code object and a file name
assert ft.editors_helpers.check_syntax(source=bad_code_exec)
assert ft.editors_helpers.check_syntax(source=good_code)
assert not ft.get_output() # no new exceptions recorded
try:
exec(bad_code_syntax, {})
except Exception:
assert not ft.get_output()
# Ensure that a call to check_syntax only install() temporarily
# if it was not installed before.
ft.uninstall()
ft.editors_helpers.check_syntax(source=bad_code_syntax)
assert not ft.is_installed()
ft.editors_helpers.check_syntax(source=bad_code_syntax, include="python_tb")
assert not ft.is_installed()
# When friendly is "installed", a call to check_syntax
# leaves its include unchanged.
ft.install(redirect="capture")
ft.set_include("explain")
ft.editors_helpers.check_syntax(source=bad_code_syntax)
assert ft.get_include() == "explain"
ft.editors_helpers.check_syntax(source=bad_code_syntax, include="python_tb")
assert ft.get_include() == "explain"
# A call to advanced_code_syntax, with a language specified as an argument
# should leave the previous language unchanged.
ft.set_lang("en")
assert not ft.editors_helpers.check_syntax(source=bad_code_syntax, lang="fr")
result = ft.get_output()
assert "SyntaxError" in result
assert ft.get_lang() == "en"
# Clean up and restore for other tests
ft.get_output()
ft.set_stream(None)
if installed:
ft.uninstall()
ft.set_include(original_include)
def test_no_tb_does_not_contain_friendly_tb_items():
"""Verify items from ``"friendly_tb"`` are discarded in ``"no_tb"`` group."""
# Strictly speaking, this is not a unit test for check_syntax proper but
# this "include" option is only used in check_syntax.
no_tb_group = ft.base_formatters.items_groups["no_tb"]
friendly_tb_group = ft.base_formatters.items_groups["friendly_tb"]
assert not any(item in no_tb_group for item in friendly_tb_group)
if __name__ == "__main__":
test_check_syntax()
print("Success!")
|
