Added matplotlib_baseline_image_generation command line option to run only tests having this option

SidharthBansal · SidharthBansal · commit 2aba426334c9 · 2020-07-09T20:41:20.000+05:30
diff --git a/lib/matplotlib/tests/conftest.py b/lib/matplotlib/tests/conftest.py
@@ -2,3 +2,25 @@
                                          mpl_image_comparison_parameters,
                                          pytest_configure, pytest_unconfigure,
                                          pd)
+import pytest
+
+
+def pytest_addoption(parser):
+    parser.addoption(
+        "--matplotlib_baseline_image_generation",
+         action="store_true",
+         default=False,
+         help="run matplotlib baseline image generation tests"
+    )
+
+
+def pytest_configure(config):
+    config.addinivalue_line("markers", "matplotlib_baseline_image_generation: mark test as image generation test")
+
+
+def pytest_collection_modifyitems(config, items):
+    if config.getoption("--matplotlib_baseline_image_generation"):
+        skip_non_matplotlib_baseline_image_generation_tests = pytest.mark.skip(reason="No ned to run non image generation tests")
+        for item in items:
+           if "matplotlib_baseline_image_generation" not in item.keywords:
+               item.add_marker(skip_non_matplotlib_baseline_image_generation_tests)
diff --git a/lib/matplotlib/tests/test_agg.py b/lib/matplotlib/tests/test_agg.py
@@ -80,7 +80,7 @@ def test_long_path():
     fig.savefig(buff, format='png')
 
 
-@pytest.mark.baseline_image_generation_test
+@pytest.mark.matplotlib_baseline_image_generation
 @image_comparison(['agg_filter.png'], remove_text=True)
 def test_agg_filter():
     def smooth1d(x, window_len):
diff --git a/pytest.ini b/pytest.ini
@@ -8,5 +8,5 @@ junit_family = xunit2
 
 
 markers =
-    baseline_image_generation_test: marks tests as baseline_image_generation_test (deselect with '-m "not baseline_image_generation_test"')
+    matplotlib_baseline_image_generation: marks tests as matplotlib_baseline_image_generation (deselect with '-m "not matplotlib_baseline_image_generation"')
     serial
