Add support of nose setup/teardown to pytest collector

Author: Kojoley

conftest.py

@@ -3,13 +3,19 @@
 
 import inspect
 import pytest
+import unittest
 
 import matplotlib
 matplotlib.use('agg')
 
 from matplotlib.testing.decorators import ImageComparisonTest
 
 
+def is_nose_class(cls):
+    return any(name in ['setUp', 'tearDown']
+               for name, _ in inspect.getmembers(cls))
+
+
 def pytest_configure(config):
     matplotlib._called_from_pytest = True
 
@@ -25,3 +31,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)