Added logic for creation of the baseline images

SidharthBansal · SidharthBansal · commit 678b7787e531 · 2020-06-30T20:33:06.000+05:30
diff --git a/.gitignore b/.gitignore
@@ -104,3 +104,7 @@ lib/z.lib
 #########################
 lib/matplotlib/backends/web_backend/node_modules/
 lib/matplotlib/backends/web_backend/package-lock.json
+
+# Matplotlib tempenv #
+######################
+tempenv
diff --git a/lib/matplotlib/testing/decorators.py b/lib/matplotlib/testing/decorators.py
@@ -6,6 +6,7 @@
 from pathlib import Path
 import shutil
 import string
+import subprocess
 import sys
 import unittest
 import warnings
@@ -486,7 +487,7 @@ def _image_directories(func):
         try:
             import matplotlib_baseline_images
         except:
-            raise ImportError("Not able to import matplotlib_baseline_images")
+            generate_matplotlib_baseline_images_from_previous_commit()
         baseline_dir = (Path(matplotlib_baseline_images.__file__).parent /
                             module_path.stem)
     elif func.__module__.startswith("mpl_toolkits."):
@@ -504,3 +505,44 @@ def _image_directories(func):
     result_dir = Path().resolve() / "result_images" / module_path.stem
     result_dir.mkdir(parents=True, exist_ok=True)
     return baseline_dir, result_dir
+
+
+def generate_matplotlib_baseline_images_from_previous_commit():
+    # Check if current repository is a git repository
+    subprocess.call(['git', '-C', '.', 'status'],
+                            stderr=subprocess.STDOUT,
+                            stdout = open(os.devnull,
+                                            'w')) == 0
+    # Check that git is installed.
+    subprocess.call(shutil.which('git'),
+                    stderr=subprocess.STDOUT,
+                    stdout = open(os.devnull,
+                                    'w')) == 0
+    # Create virtual env in tmpdir.
+    subprocess.run(["virtualenv", "tmpenv"],
+                    stderr=subprocess.STDOUT,
+                    stdout=subprocess.DEVNULL))
+    # Clone mpl repo to tmpvenv and run pytests from new mpl repo created
+    subprocess.run(["git", "clone", "../../..", "../../../tmpenv"],
+                    stderr=subprocess.STDOUT,
+                    stdout=subprocess.DEVNULL))
+    subprocess.run(["cd", "../../../tmpenv"],
+                    stderr=subprocess.STDOUT,
+                    stdout=subprocess.DEVNULL))
+    subprocess.run([".", ".tempenv/bin/activate"],
+                    stderr=subprocess.STDOUT,
+                    stdout=subprocess.DEVNULL))
+    subprocess.run(["git", "checkout", "HEAD~"],
+                    stderr=subprocess.STDOUT,
+                    stdout=subprocess.DEVNULL))
+    subprocess.run(["python", "-mpip", "install", "-e", "."],
+                    stderr=subprocess.STDOUT,
+                    stdout=subprocess.DEVNULL))
+    subprocess.run(["python", "-mpytest"],
+                    stderr=subprocess.STDOUT,
+                    stdout=subprocess.DEVNULL))
+    # Copy result_images in the temporary directory and cache it
+    subprocess.run(["cp, "-r", "../../../result_images",
+                    "../../../tempenv/.pytest_cache"],
+                    stderr=subprocess.STDOUT,
+                    stdout=subprocess.DEVNULL))
