support @pytest.yield_fixture

Author: reverbc

pylint_pytest.py

@@ -33,7 +33,8 @@ def _is_pytest_fixture(decorator):
             # expecting @pytest.fixture(scope=...)
             attr = decorator.func
 
-        if attr and attr.attrname == 'fixture' and attr.expr.name == 'pytest':
+        if attr and attr.attrname in ('fixture', 'yield_fixture') \
+                and attr.expr.name == 'pytest':
             return True
     except AttributeError:
         pass

tests/input/redefined-outer-name/caller_yield_fixture.py

@@ -0,0 +1,7 @@
+import pytest
+from conftest import conftest_fixture_attr
+
+
+@pytest.yield_fixture
+def caller_yield_fixture(conftest_fixture_attr):
+    assert conftest_fixture_attr

tests/input/unused-argument/caller_yield_fixture.py

@@ -0,0 +1,7 @@
+import pytest
+from conftest import conftest_fixture_attr
+
+
+@pytest.yield_fixture
+def caller_yield_fixture(conftest_fixture_attr):
+    assert True

tests/input/unused-import/caller_yield_fixture.py

@@ -0,0 +1,7 @@
+import pytest
+from conftest import conftest_fixture_attr
+
+
+@pytest.yield_fixture
+def caller_yield_fixture(conftest_fixture_attr):
+    assert conftest_fixture_attr

tests/test_redefined_outer_name.py

@@ -12,6 +12,13 @@ def test_smoke(self, enable_plugin):
             msg_count=0 if enable_plugin else 1,
         )
 
+    @pytest.mark.parametrize('enable_plugin', [True, False])
+    def test_caller_yield_fixture(self, enable_plugin):
+        self.run_test(
+            enable_plugin=enable_plugin,
+            msg_count=0 if enable_plugin else 1,
+        )
+
     @pytest.mark.parametrize('enable_plugin', [True, False])
     def test_caller_not_a_test_func(self, enable_plugin):
         self.run_test(

tests/test_unused_argument.py

@@ -12,6 +12,13 @@ def test_smoke(self, enable_plugin):
             msg_count=0 if enable_plugin else 2,
         )
 
+    @pytest.mark.parametrize('enable_plugin', [True, False])
+    def test_caller_yield_fixture(self, enable_plugin):
+        self.run_test(
+            enable_plugin=enable_plugin,
+            msg_count=0 if enable_plugin else 1,
+        )
+
     @pytest.mark.parametrize('enable_plugin', [True, False])
     def test_caller_not_a_test_func(self, enable_plugin):
         self.run_test(

tests/test_unused_import.py

@@ -12,6 +12,13 @@ def test_smoke(self, enable_plugin):
             msg_count=0 if enable_plugin else 1,
         )
 
+    @pytest.mark.parametrize('enable_plugin', [True, False])
+    def test_caller_yield_fixture(self, enable_plugin):
+        self.run_test(
+            enable_plugin=enable_plugin,
+            msg_count=0 if enable_plugin else 1,
+        )
+
     @pytest.mark.parametrize('enable_plugin', [True, False])
     def test_same_name_arg(self, enable_plugin):
         '''an unused import (not a fixture) just happened to have the same