summaryrefslogtreecommitdiff
path: root/astroid/tests/unittest_brain.py
diff options
context:
space:
mode:
Diffstat (limited to 'astroid/tests/unittest_brain.py')
-rw-r--r--astroid/tests/unittest_brain.py22
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()