Fix interference with pytest-run-parallel detection of known thread-unsafe calls

Author: ayshih

pytest_arraydiff/plugin.py

@@ -266,10 +266,6 @@ def __init__(self, config, reference_dir=None, generate_dir=None, default_format
         self.default_format = default_format
         self.return_value = {}
 
-    def pytest_collection_modifyitems(self, items):
-        for item in items:
-            wrap_array_interceptor(self, item)
-
     @pytest.hookimpl(hookwrapper=True)
     def pytest_runtest_call(self, item):
 
@@ -308,6 +304,8 @@ def pytest_runtest_call(self, item):
 
         baseline_remote = reference_dir.startswith('http')
 
+        # Run test and get array object
+        wrap_array_interceptor(self, item)
         yield
         test_name = generate_test_name(item)
         if test_name not in self.return_value:
@@ -380,6 +378,11 @@ def __init__(self, config):
         self.config = config
         self.return_value = {}
 
-    def pytest_collection_modifyitems(self, items):
-        for item in items:
+    @pytest.hookimpl(hookwrapper=True)
+    def pytest_runtest_call(self, item):
+
+        if item.get_closest_marker('array_compare') is not None:
             wrap_array_interceptor(self, item)
+
+        yield
+        return