Added logic for creation of baseline images

SidharthBansal · SidharthBansal · commit a000051fb21a · 2020-07-01T19:11:59.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 #
+######################
+temp/testenv
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,15 +487,14 @@ 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."):
         try:
             import mpl_toolkits_baseline_images
         except:
-            raise ImportError("Not able to import "
-                              "mpl_toolkits_baseline_images")
+            generate_matplotlib_baseline_images_from_previous_commit()
         baseline_dir = (Path(mpl_toolkits_baseline_images.__file__).parent /
                             module_path.stem)
     else:
@@ -504,3 +504,62 @@ 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():
+    mplRepoPath = Path(mpl.__file__).parent
+    # Check that git is installed.
+    subprocess.run(shutil.which('git'),
+                    stderr=subprocess.STDOUT,
+                    stdout = open(os.devnull, 'w'),
+                    universal_newlines=True) == None
+    # Check if current repository is a git repository
+    subprocess.run(['git', 'status'],
+                    cwd = mplRepoPath,
+                    stderr=subprocess.STDOUT,
+                    stdout = open(os.devnull, 'w'),
+                    universal_newlines=True) == None
+    # Create virtual env in tmpdir.
+#    if not (mplRepoPath / 'tmp/testenv').exists():
+#        modify_existing_baseline_images()
+#    else
+        subprocess.run(["python", "-mvenv", "tmp/testenv"],
+                        cwd = mplRepoPath,
+                        stderr=subprocess.STDOUT,
+                        stdout=subprocess.DEVNULL),
+                        universal_newlines=True)
+        # Clone mpl repo to tmpvenv and run pytests from new mpl repo created
+        subprocess.run(["git", "clone", str(mplRepoPath), "tmp/testenv"],
+                        cwd = mplRepoPath,
+                        stderr=subprocess.STDOUT,
+                        stdout=subprocess.DEVNULL,
+                        universal_newlines=True)
+        subprocess.run(["tmp/testenv/bin/python"],
+                        cwd = mplRepoPath,
+                        stderr=subprocess.STDOUT,
+                        stdout=subprocess.DEVNULL,
+                        universal_newlines=True)
+        subprocess.run(["git", "checkout", "HEAD~"],
+                        cwd = mplRepoPath,
+                        stderr=subprocess.STDOUT,
+                        stdout=subprocess.DEVNULL,
+                        universal_newlines=True)
+        subprocess.run(["python", "-mpip", "install", "-e", mplRepoPath],
+                        cwd = mplRepoPath,
+                        stderr=subprocess.STDOUT,
+                        stdout=subprocess.DEVNULL,
+                        universal_newlines=True)
+        subprocess.run(["python", "-mpytest"],
+                        cwd = mplRepoPath,
+                        stderr=subprocess.STDOUT,
+                        stdout=subprocess.DEVNULL,
+                        universal_newlines=True)
+        # Copy result_images in the temporary directory and cache it
+        subprocess.run(shutil.copytree("result_images",
+                                        "tempenv/.pytest_cache"),
+                        cwd = mplRepoPath,
+                        stderr=subprocess.STDOUT,
+                        stdout=subprocess.DEVNUL,
+                        universal_newlines=True)
+
+# def modify_existing_baseline_images():
