Tim Peters | fc35de4 | 2001-03-02 01:48:16 +0000 | [diff] [blame] | 1 | """Verify that warnings are issued for global statements following use.""" |
Jeremy Hylton | 8e43cd7 | 2001-02-28 01:51:01 +0000 | [diff] [blame] | 2 | |
Brett Cannon | 2298d53 | 2010-03-20 21:48:19 +0000 | [diff] [blame] | 3 | from test.support import run_unittest, check_syntax_error, check_warnings |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 4 | import unittest |
Jeremy Hylton | 8e43cd7 | 2001-02-28 01:51:01 +0000 | [diff] [blame] | 5 | import warnings |
Brett Cannon | 2298d53 | 2010-03-20 21:48:19 +0000 | [diff] [blame] | 6 | |
Jeremy Hylton | 8e43cd7 | 2001-02-28 01:51:01 +0000 | [diff] [blame] | 7 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 8 | class GlobalTests(unittest.TestCase): |
Jeremy Hylton | 8e43cd7 | 2001-02-28 01:51:01 +0000 | [diff] [blame] | 9 | |
Brett Cannon | 2298d53 | 2010-03-20 21:48:19 +0000 | [diff] [blame] | 10 | def setUp(self): |
| 11 | self._warnings_manager = check_warnings() |
| 12 | self._warnings_manager.__enter__() |
| 13 | warnings.filterwarnings("error", module="<test string>") |
| 14 | |
| 15 | def tearDown(self): |
| 16 | self._warnings_manager.__exit__(None, None, None) |
| 17 | |
| 18 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 19 | def test1(self): |
| 20 | prog_text_1 = """\ |
Jeremy Hylton | 8e43cd7 | 2001-02-28 01:51:01 +0000 | [diff] [blame] | 21 | def wrong1(): |
| 22 | a = 1 |
| 23 | b = 2 |
| 24 | global a |
| 25 | global b |
| 26 | """ |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 27 | check_syntax_error(self, prog_text_1) |
Jeremy Hylton | 8e43cd7 | 2001-02-28 01:51:01 +0000 | [diff] [blame] | 28 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 29 | def test2(self): |
| 30 | prog_text_2 = """\ |
Jeremy Hylton | 8e43cd7 | 2001-02-28 01:51:01 +0000 | [diff] [blame] | 31 | def wrong2(): |
Neal Norwitz | 752abd0 | 2008-05-13 04:55:24 +0000 | [diff] [blame] | 32 | print(x) |
Jeremy Hylton | 8e43cd7 | 2001-02-28 01:51:01 +0000 | [diff] [blame] | 33 | global x |
| 34 | """ |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 35 | check_syntax_error(self, prog_text_2) |
Jeremy Hylton | 8e43cd7 | 2001-02-28 01:51:01 +0000 | [diff] [blame] | 36 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 37 | def test3(self): |
| 38 | prog_text_3 = """\ |
Jeremy Hylton | 8e43cd7 | 2001-02-28 01:51:01 +0000 | [diff] [blame] | 39 | def wrong3(): |
Neal Norwitz | 752abd0 | 2008-05-13 04:55:24 +0000 | [diff] [blame] | 40 | print(x) |
Jeremy Hylton | 8e43cd7 | 2001-02-28 01:51:01 +0000 | [diff] [blame] | 41 | x = 2 |
| 42 | global x |
| 43 | """ |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 44 | check_syntax_error(self, prog_text_3) |
Jeremy Hylton | 2922ea8 | 2001-02-28 23:49:19 +0000 | [diff] [blame] | 45 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 46 | def test4(self): |
| 47 | prog_text_4 = """\ |
Jeremy Hylton | 2922ea8 | 2001-02-28 23:49:19 +0000 | [diff] [blame] | 48 | global x |
| 49 | x = 2 |
| 50 | """ |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 51 | # this should work |
| 52 | compile(prog_text_4, "<test string>", "exec") |
| 53 | |
| 54 | |
| 55 | def test_main(): |
Florent Xicluna | 41fe615 | 2010-04-02 18:52:12 +0000 | [diff] [blame] | 56 | with warnings.catch_warnings(): |
| 57 | warnings.filterwarnings("error", module="<test string>") |
| 58 | run_unittest(GlobalTests) |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 59 | |
| 60 | if __name__ == "__main__": |
| 61 | test_main() |