flake8/flake8/tests/test_flakes.py
Ian Cordasco 25dd8c445e Fix tests
2013-02-10 13:55:57 -05:00

61 lines
1 KiB
Python

from unittest import TestCase
from pyflakes.api import check
code = """
try:
pass
except ValueError as err:
print(err)
"""
code2 = """
try:
pass
except ValueError:
print("err")
try:
pass
except ValueError:
print("err")
"""
code3 = """
try:
pass
except (ImportError, ValueError):
print("err")
"""
code_from_import_exception = """
from foo import SomeException
try:
pass
except SomeException:
print("err")
"""
code_import_exception = """
import foo.SomeException
try:
pass
except foo.SomeException:
print("err")
"""
class TestFlake(TestCase):
def test_exception(self):
for c in (code, code2, code3):
warnings = check(code, '(stdin)')
self.assertEqual(warnings, 0)
def test_from_import_exception_in_scope(self):
warnings = check(code_from_import_exception, '(stdin)')
self.assertEqual(warnings, 0)
def test_import_exception_in_scope(self):
warnings = check(code_import_exception, '(stdin)')
self.assertEqual(warnings, 0)