Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 7 additions & 8 deletions pytest_mpl/plugin.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,20 +39,12 @@

import pytest

import matplotlib
import matplotlib.pyplot as plt
from matplotlib.testing.compare import compare_images
from matplotlib.testing.decorators import ImageComparisonTest as MplImageComparisonTest

if sys.version_info[0] == 2:
from urllib import urlopen
else:
from urllib.request import urlopen


MPL_LT_15 = LooseVersion(matplotlib.__version__) < LooseVersion('1.5')


def _download_file(url):
u = urlopen(url)
result_dir = tempfile.mkdtemp()
Expand Down Expand Up @@ -115,6 +107,13 @@ def __init__(self, config, baseline_dir=None, generate_dir=None, results_dir=Non

def pytest_runtest_setup(self, item):

import matplotlib
import matplotlib.pyplot as plt
from matplotlib.testing.compare import compare_images
from matplotlib.testing.decorators import ImageComparisonTest as MplImageComparisonTest

MPL_LT_15 = LooseVersion(matplotlib.__version__) < LooseVersion('1.5')

compare = item.keywords.get('mpl_image_compare')

if compare is None:
Expand Down