diff options
Diffstat (limited to 'astroid/tests/unittest_brain.py')
-rw-r--r-- | astroid/tests/unittest_brain.py | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/astroid/tests/unittest_brain.py b/astroid/tests/unittest_brain.py index 861fab2..cb35462 100644 --- a/astroid/tests/unittest_brain.py +++ b/astroid/tests/unittest_brain.py @@ -59,6 +59,12 @@ try: except ImportError: HAS_NUMPY = False +try: + import pytest # pylint: disable=unused-import + HAS_PYTEST = True +except ImportError: + HAS_PYTEST = False + class HashlibTest(unittest.TestCase): def test_hashlib(self): @@ -450,5 +456,21 @@ class NumpyBrainTest(unittest.TestCase): self.assertIsInstance(inferred, nodes.FunctionDef) +@unittest.skipUnless(HAS_PYTEST, "This test requires the pytest library.") +class PytestBrainTest(unittest.TestCase): + + def test_pytest(self): + ast_node = test_utils.extract_node(''' + import pytest + pytest #@ + ''') + module = next(ast_node.infer()) + self.assertIn('deprecated_call', module) + self.assertIn('exit', module) + self.assertIn('fail', module) + self.assertIn('fixture', module) + self.assertIn('mark', module) + + if __name__ == '__main__': unittest.main() |