Add support of nose setup/teardown to pytest collector

Kojoley · Kojoley · commit 8b64294b667a · 2016-07-24T01:13:00.000+03:00
diff --git a/conftest.py b/conftest.py
@@ -3,11 +3,18 @@
 
 import inspect
 import pytest
+import unittest
 
 import matplotlib
 matplotlib.use('agg')
 
 from matplotlib.testing.decorators import ImageComparisonTest
+from matplotlib import default_test_modules
+
+
+def is_nose_class(cls):
+    return any(name in ['setUp', 'tearDown']
+               for name, _ in inspect.getmembers(cls))
 
 
 def pytest_configure(config):
@@ -25,3 +32,20 @@ def pytest_pycollect_makeitem(collector, name, obj):
             # instead of function and this confuses pytest because it crawls
             # original names and sees 'test_*', but not 'Test*' in that case
             return pytest.Class(name, parent=collector)
+
+        if is_nose_class(obj) and not issubclass(obj, unittest.TestCase):
+            # Workaround unittest-like setup/teardown names in pure classes
+            setup = getattr(obj, 'setUp', None)
+            if setup is not None:
+                obj.setup_method = lambda self, _: obj.setUp(self)
+            tearDown = getattr(obj, 'tearDown', None)
+            if tearDown is not None:
+                obj.teardown_method = lambda self, _: obj.tearDown(self)
+            setUpClass = getattr(obj, 'setUpClass', None)
+            if setUpClass is not None:
+                obj.setup_class = obj.setUpClass
+            tearDownClass = getattr(obj, 'tearDownClass', None)
+            if tearDownClass is not None:
+                obj.teardown_class = obj.tearDownClass
+
+            return pytest.Class(name, parent=collector)
