|
1 | 1 | from matplotlib.testing.noseclasses import KnownFailureTest, \ |
2 | 2 | KnownFailureDidNotFailTest, ImageComparisonFailure |
3 | | -import os, sys |
| 3 | +import os, sys, shutil |
4 | 4 | import nose |
5 | 5 | import matplotlib |
6 | 6 | import matplotlib.tests |
@@ -85,7 +85,11 @@ def compare_images_decorator(func): |
85 | 85 |
|
86 | 86 | def compare_images_generator(): |
87 | 87 | for extension in extensions: |
88 | | - expected_fnames = [os.path.join(baseline_dir,fname) + '.' + extension for fname in baseline_images] |
| 88 | + orig_expected_fnames = [os.path.join(baseline_dir,fname) + '.' + extension for fname in baseline_images] |
| 89 | + expected_fnames = [os.path.join(result_dir,'expected-'+fname) + '.' + extension for fname in baseline_images] |
| 90 | + for src,dst in zip( orig_expected_fnames, expected_fnames ): |
| 91 | + if not os.path.exists(dst): |
| 92 | + shutil.copyfile(src,dst) |
89 | 93 | actual_fnames = [os.path.join(result_dir, fname) + '.' + extension for fname in baseline_images] |
90 | 94 | have_baseline_images = [os.path.exists(expected) for expected in expected_fnames] |
91 | 95 | have_baseline_image = np.all(have_baseline_images) |
@@ -148,15 +152,10 @@ def _image_directories(func): |
148 | 152 | basedir = os.path.dirname(matplotlib.tests.__file__) |
149 | 153 |
|
150 | 154 | baseline_dir = os.path.join(basedir,'baseline_images',subdir) |
151 | | - result_dir = os.path.join(basedir,'current_images',subdir) |
| 155 | + result_dir = os.path.abspath(os.path.join('result_images',subdir)) |
152 | 156 |
|
153 | 157 | if not os.path.exists(result_dir): |
154 | | - try: |
155 | | - # make the current_images directory first |
156 | | - os.mkdir(os.path.join(basedir,'current_images')) |
157 | | - except OSError: |
158 | | - pass # probably exists already |
159 | | - os.mkdir(result_dir) |
| 158 | + os.makedirs(result_dir) |
160 | 159 |
|
161 | 160 | return baseline_dir, result_dir |
162 | 161 |
|
0 commit comments