from __future__ import print_function, absolute_import

from importlib import import_module

from distutils import sysconfig

from distutils import version

from distutils.core import Extension

import hashlib

import distutils.command.build_ext

from io import BytesIO

import glob

import multiprocessing

import os

import platform

import re

import setuptools

import shutil

import subprocess

from subprocess import check_output

import sys

import warnings

from textwrap import fill

import shutil

import tarfile

import textwrap

import versioneer

import warnings

import contextlib

if sys.version_info < (3, ):

from urllib2 import urlopen, Request

class FileExistsError(OSError):

pass

def makedirs(path, exist_ok=True):

if not exist_ok:

raise ValueError("this backport only supports exist_ok is True")

if not path or os.path.exists(path):

return

head, tail = os.path.split(path)

makedirs(head, exist_ok=True)

os.makedirs(path)

else:

from urllib.request import urlopen, Request

from os import makedirs

PY3min = (sys.version_info[0] >= 3)

def _get_xdg_cache_dir():

"""

Return the XDG cache directory.

See https://standards.freedesktop.org/basedir-spec/basedir-spec-latest.html

"""

cache_dir = os.environ.get('XDG_CACHE_HOME')

if not cache_dir:

cache_dir = os.path.expanduser('~/.cache')

if cache_dir.startswith('~/'): # Expansion failed.

return None

return os.path.join(cache_dir, 'matplotlib')

def get_fd_hash(fd):

"""

Compute the sha256 hash of the bytes in a file-like

"""

BLOCKSIZE = 1 << 16

hasher = hashlib.sha256()

old_pos = fd.tell()

fd.seek(0)

buf = fd.read(BLOCKSIZE)

while buf:

hasher.update(buf)

buf = fd.read(BLOCKSIZE)

fd.seek(old_pos)

return hasher.hexdigest()

def download_or_cache(url, sha):

"""

Get bytes from the given url or local cache.

Parameters

----------

url : str

The url to download

sha : str

The sha256 of the file

Returns

-------

BytesIO

The file loaded into memory.

"""

cache_dir = _get_xdg_cache_dir()

def get_from_cache(local_fn):

if cache_dir is None:

raise Exception("no cache dir")

cache_filename = os.path.join(cache_dir, local_fn)

with open(cache_filename, 'rb') as fin:

buf = BytesIO(fin.read())

file_sha = get_fd_hash(buf)

if file_sha != sha:

return None

buf.seek(0)

return buf

def write_cache(local_fn, data):

if cache_dir is None:

raise Exception("no cache dir")

cache_filename = os.path.join(cache_dir, local_fn)

makedirs(cache_dir, exist_ok=True)

if sys.version_info < (3, ):

if os.path.exists(cache_filename):

raise FileExistsError

mode = 'wb'

else:

mode = 'xb'

old_pos = data.tell()

data.seek(0)

with open(cache_filename, mode=mode) as fout:

fout.write(data.read())

data.seek(old_pos)

try:

return get_from_cache(sha)

except Exception:

pass

# jQueryUI's website blocks direct downloads from urllib.request's

# default User-Agent, but not (for example) wget; so I don't feel too

# bad passing in an empty User-Agent.

with contextlib.closing(urlopen(

Request(url, headers={"User-Agent": ""}))) as req:

file_contents = BytesIO(req.read())

file_contents.seek(0)

file_sha = get_fd_hash(file_contents)

if file_sha != sha:

raise Exception(("The download file does not match the "

"expected sha. {url} was expected to have "

"{sha} but it had {file_sha}").format(

sha=sha, file_sha=file_sha, url=url))

try:

write_cache(sha, file_contents)

except Exception:

pass

file_contents.seek(0)

return file_contents

# SHA256 hashes of the FreeType tarballs

_freetype_hashes = {

'2.6.1': '0a3c7dfbda6da1e8fce29232e8e96d987ababbbf71ebc8c75659e4132c367014',

'2.6.2': '8da42fc4904e600be4b692555ae1dcbf532897da9c5b9fb5ebd3758c77e5c2d4',

'2.6.3': '7942096c40ee6fea882bd4207667ad3f24bff568b96b10fd3885e11a7baad9a3',

'2.6.4': '27f0e38347a1850ad57f84fc4dfed68ba0bc30c96a6fa6138ef84d485dd9a8d7',

'2.6.5': '3bb24add9b9ec53636a63ea8e867ed978c4f8fdd8f1fa5ccfd41171163d4249a',

'2.7': '7b657d5f872b0ab56461f3bd310bd1c5ec64619bd15f0d8e08282d494d9cfea4',

'2.7.1': '162ef25aa64480b1189cdb261228e6c5c44f212aac4b4621e28cf2157efb59f5',

'2.8': '33a28fabac471891d0523033e99c0005b95e5618dc8ffa7fa47f9dadcacb1c9b',

'2.8.1': '876711d064a6a1bd74beb18dd37f219af26100f72daaebd2d86cb493d7cd7ec6',

}

# This is the version of FreeType to use when building a local

# version. It must match the value in

# lib/matplotlib.__init__.py and also needs to be changed below in the

# embedded windows build script (grep for "REMINDER" in this file)

LOCAL_FREETYPE_VERSION = '2.6.1'

LOCAL_FREETYPE_HASH = _freetype_hashes.get(LOCAL_FREETYPE_VERSION, 'unknown')

if sys.platform != 'win32':

if not PY3min:

from commands import getstatusoutput

else:

from subprocess import getstatusoutput

if PY3min:

import configparser

else:

import ConfigParser as configparser

# matplotlib build options, which can be altered using setup.cfg

options = {

'display_status': True,

'verbose': False,

'backend': None,

'basedirlist': None

}

setup_cfg = os.environ.get('MPLSETUPCFG', 'setup.cfg')

if os.path.exists(setup_cfg):

if PY3min:

config = configparser.ConfigParser()

else:

config = configparser.SafeConfigParser()

config.read(setup_cfg)

if config.has_option('status', 'suppress'):

options['display_status'] = not config.getboolean("status", "suppress")

if config.has_option('rc_options', 'backend'):

options['backend'] = config.get("rc_options", "backend")

if config.has_option('directories', 'basedirlist'):

options['basedirlist'] = [

x.strip() for x in

config.get("directories", "basedirlist").split(',')]

if config.has_option('test', 'local_freetype'):

options['local_freetype'] = config.getboolean("test", "local_freetype")

else:

config = None

lft = bool(os.environ.get('MPLLOCALFREETYPE', False))

options['local_freetype'] = lft or options.get('local_freetype', False)

def get_win32_compiler():

"""

Determine the compiler being used on win32.

"""

# Used to determine mingw32 or msvc

# This is pretty bad logic, someone know a better way?

for v in sys.argv:

if 'mingw32' in v:

return 'mingw32'

return 'msvc'

win32_compiler = get_win32_compiler()

def extract_versions():

"""

Extracts version values from the main matplotlib __init__.py and

returns them as a dictionary.

"""

with open('lib/matplotlib/__init__.py') as fd:

for line in fd.readlines():

if (line.startswith('__version__numpy__')):

exec(line.strip())

return locals()

def has_include_file(include_dirs, filename):

"""

Returns `True` if `filename` can be found in one of the

directories in `include_dirs`.

"""

if sys.platform == 'win32':

include_dirs = list(include_dirs) # copy before modify

include_dirs += os.environ.get('INCLUDE', '.').split(os.pathsep)

for dir in include_dirs:

if os.path.exists(os.path.join(dir, filename)):

return True

return False

def check_include_file(include_dirs, filename, package):

"""

Raises an exception if the given include file can not be found.

"""

if not has_include_file(include_dirs, filename):

raise CheckFailed(

"The C/C++ header for %s (%s) could not be found. You "

"may need to install the development package." %

(package, filename))

def get_base_dirs():

"""

Returns a list of standard base directories on this platform.

"""

if options['basedirlist']:

return options['basedirlist']

if os.environ.get('MPLBASEDIRLIST'):

return os.environ.get('MPLBASEDIRLIST').split(os.pathsep)

win_bases = ['win32_static', ]

# on conda windows, we also add the <conda_env_dir>\Library,

# as conda installs libs/includes there

# env var names mess: https://github.com/conda/conda/issues/2312

conda_env_path = os.getenv('CONDA_PREFIX') # conda >= 4.1

if not conda_env_path:

conda_env_path = os.getenv('CONDA_DEFAULT_ENV') # conda < 4.1

if conda_env_path and os.path.isdir(conda_env_path):

win_bases.append(os.path.join(conda_env_path, "Library"))

basedir_map = {

'win32': win_bases,

'darwin': ['/usr/local/', '/usr', '/usr/X11',

'/opt/X11', '/opt/local'],

'sunos5': [os.getenv('MPLIB_BASE') or '/usr/local', ],

'gnu0': ['/usr'],

'aix5': ['/usr/local'],

}

return basedir_map.get(sys.platform, ['/usr/local', '/usr'])

def get_include_dirs():

"""

Returns a list of standard include directories on this platform.

"""

include_dirs = [os.path.join(d, 'include') for d in get_base_dirs()]

if sys.platform != 'win32':

# gcc includes this dir automatically, so also look for headers in

# these dirs

include_dirs.extend(

os.environ.get('CPLUS_INCLUDE_PATH', '').split(os.pathsep))

return include_dirs

def is_min_version(found, minversion):

"""

Returns `True` if `found` is at least as high a version as

`minversion`.

"""

expected_version = version.LooseVersion(minversion)

found_version = version.LooseVersion(found)

return found_version >= expected_version

# Define the display functions only if display_status is True.

if options['display_status']:

def print_line(char='='):

print(char * 76)

def print_status(package, status):

initial_indent = "%22s: " % package

indent = ' ' * 24

print(fill(str(status), width=76,

initial_indent=initial_indent,

subsequent_indent=indent))

def print_message(message):

indent = ' ' * 24 + "* "

print(fill(str(message), width=76,

initial_indent=indent,

subsequent_indent=indent))

def print_raw(section):

print(section)

else:

def print_line(*args, **kwargs):

pass

print_status = print_message = print_raw = print_line

# Remove the -Wstrict-prototypes option, is it's not valid for C++

customize_compiler = distutils.command.build_ext.customize_compiler

def my_customize_compiler(compiler):

retval = customize_compiler(compiler)

try:

compiler.compiler_so.remove('-Wstrict-prototypes')

except (ValueError, AttributeError):

pass

return retval

distutils.command.build_ext.customize_compiler = my_customize_compiler

def make_extension(name, files, *args, **kwargs):

"""

Make a new extension. Automatically sets include_dirs and

library_dirs to the base directories appropriate for this

platform.

`name` is the name of the extension.

`files` is a list of source files.

Any additional arguments are passed to the

`distutils.core.Extension` constructor.

"""

ext = DelayedExtension(name, files, *args, **kwargs)

for dir in get_base_dirs():

include_dir = os.path.join(dir, 'include')

if os.path.exists(include_dir):

ext.include_dirs.append(include_dir)

for lib in ('lib', 'lib64'):

lib_dir = os.path.join(dir, lib)

if os.path.exists(lib_dir):

ext.library_dirs.append(lib_dir)

ext.include_dirs.append('.')

return ext

def get_buffer_hash(fd):

BLOCKSIZE = 1 << 16

hasher = hashlib.sha256()

buf = fd.read(BLOCKSIZE)

while buf:

hasher.update(buf)

buf = fd.read(BLOCKSIZE)

return hasher.hexdigest()

class PkgConfig(object):

"""

This is a class for communicating with pkg-config.

"""

def __init__(self):

"""

Determines whether pkg-config exists on this machine.

"""

if sys.platform == 'win32':

self.has_pkgconfig = False

else:

try:

self.pkg_config = os.environ['PKG_CONFIG']

except KeyError:

self.pkg_config = 'pkg-config'

self.set_pkgconfig_path()

status, output = getstatusoutput(self.pkg_config + " --help")

self.has_pkgconfig = (status == 0)

if not self.has_pkgconfig:

print("IMPORTANT WARNING:")

print(

" pkg-config is not installed.\n"

" matplotlib may not be able to find some of its dependencies")

def set_pkgconfig_path(self):

pkgconfig_path = sysconfig.get_config_var('LIBDIR')

if pkgconfig_path is None:

return

pkgconfig_path = os.path.join(pkgconfig_path, 'pkgconfig')

if not os.path.isdir(pkgconfig_path):

return

try:

os.environ['PKG_CONFIG_PATH'] += ':' + pkgconfig_path

except KeyError:

os.environ['PKG_CONFIG_PATH'] = pkgconfig_path

def setup_extension(self, ext, package, default_include_dirs=[],

default_library_dirs=[], default_libraries=[],

alt_exec=None):

"""

Add parameters to the given `ext` for the given `package`.

"""

flag_map = {

'-I': 'include_dirs', '-L': 'library_dirs', '-l': 'libraries'}

executable = alt_exec

if self.has_pkgconfig:

executable = (self.pkg_config + ' {0}').format(package)

use_defaults = True

if executable is not None:

command = "{0} --libs --cflags ".format(executable)

try:

output = check_output(command, shell=True,

stderr=subprocess.STDOUT)

except subprocess.CalledProcessError:

pass

else:

output = output.decode(sys.getfilesystemencoding())

use_defaults = False

for token in output.split():

attr = flag_map.get(token[:2])

if attr is not None:

getattr(ext, attr).insert(0, token[2:])

if use_defaults:

basedirs = get_base_dirs()

for base in basedirs:

for include in default_include_dirs:

dir = os.path.join(base, include)

if os.path.exists(dir):

ext.include_dirs.append(dir)

for lib in default_library_dirs:

dir = os.path.join(base, lib)

if os.path.exists(dir):

ext.library_dirs.append(dir)

ext.libraries.extend(default_libraries)

return True

return False

def get_version(self, package):

"""

Get the version of the package from pkg-config.

"""

if not self.has_pkgconfig:

return None

status, output = getstatusoutput(

self.pkg_config + " %s --modversion" % (package))

if status == 0:

return output

return None

# The PkgConfig class should be used through this singleton

pkg_config = PkgConfig()

class CheckFailed(Exception):

"""

Exception thrown when a `SetupPackage.check` method fails.

"""

pass

class SetupPackage(object):

optional = False

pkg_names = {

"apt-get": None,

"yum": None,

"dnf": None,

"brew": None,

"port": None,

"windows_url": None

}

def check(self):

"""

Checks whether the build dependencies are met. Should raise a

`CheckFailed` exception if the dependency could not be met, otherwise

return a string indicating a version number or some other message

indicating what was found.

"""

pass

def runtime_check(self):

"""

True if the runtime dependencies of the backend are met. Assumes that

the build-time dependencies are met.

"""

return True

def get_packages(self):

"""

Get a list of package names to add to the configuration.

These are added to the `packages` list passed to

`distutils.setup`.

"""

return []

def get_namespace_packages(self):

"""

Get a list of namespace package names to add to the configuration.

These are added to the `namespace_packages` list passed to

`distutils.setup`.

"""

return []

def get_py_modules(self):

"""

Get a list of top-level modules to add to the configuration.

These are added to the `py_modules` list passed to

`distutils.setup`.

"""

return []

def get_package_data(self):

"""

Get a package data dictionary to add to the configuration.

These are merged into to the `package_data` list passed to

`distutils.setup`.

"""

return {}

def get_extension(self):

"""

Get a list of C extensions (`distutils.core.Extension`

objects) to add to the configuration. These are added to the

`extensions` list passed to `distutils.setup`.

"""

return None

def get_install_requires(self):

"""

Get a list of Python packages that we require.

pip/easy_install will attempt to download and install this

package if it is not installed.

"""

return []

def get_setup_requires(self):

"""

Get a list of Python packages that we require at build time.

pip/easy_install will attempt to download and install this

package if it is not installed.

"""

return []

def _check_for_pkg_config(self, package, include_file, min_version=None,

version=None):

"""

A convenience function for writing checks for a

pkg_config-defined dependency.

`package` is the pkg_config package name.

`include_file` is a top-level include file we expect to find.

`min_version` is the minimum version required.

`version` will override the found version if this package

requires an alternate method for that. Set version='unknown'

if the version is not known but you still want to disabled

pkg_config version check.

"""

if version is None:

version = pkg_config.get_version(package)

if version is None:

raise CheckFailed(

"pkg-config information for '%s' could not be found." %

package)

if min_version == 'PATCH':

raise CheckFailed(

"Requires patches that have not been merged upstream.")

if min_version and version != 'unknown':

if (not is_min_version(version, min_version)):

raise CheckFailed(

"Requires %s %s or later. Found %s." %

(package, min_version, version))

ext = self.get_extension()

if ext is None:

ext = make_extension('test', [])

pkg_config.setup_extension(ext, package)

check_include_file(

ext.include_dirs + get_include_dirs(), include_file, package)

return 'version %s' % version

def do_custom_build(self):

"""

If a package needs to do extra custom things, such as building a

third-party library, before building an extension, it should

override this method.

"""

pass

def install_help_msg(self):

"""

Do not override this method !

Generate the help message to show if the package is not installed.

To use this in subclasses, simply add the dictionary `pkg_names` as

a class variable:

pkg_names = {

"apt-get": <Name of the apt-get package>,

"yum": <Name of the yum package>,

"dnf": <Name of the dnf package>,

"brew": <Name of the brew package>,

"port": <Name of the port package>,

"windows_url": <The url which has installation instructions>

}

All the dictionary keys are optional. If a key is not present or has

the value `None` no message is provided for that platform.

"""

def _try_managers(*managers):

for manager in managers:

pkg_name = self.pkg_names.get(manager, None)

if pkg_name:

try:

# `shutil.which()` can be used when Python 2.7 support

# is dropped. It is available in Python 3.3+

_ = check_output(["which", manager],

stderr=subprocess.STDOUT)

if manager == 'port':

pkgconfig = 'pkgconfig'

else:

pkgconfig = 'pkg-config'

return ('Try installing {0} with `{1} install {2}` '

'and pkg-config with `{1} install {3}`'

.format(self.name, manager, pkg_name,

pkgconfig))

except subprocess.CalledProcessError:

pass

message = None

if sys.platform == "win32":

url = self.pkg_names.get("windows_url", None)

if url:

message = ('Please check {0} for instructions to install {1}'

.format(url, self.name))

elif sys.platform == "darwin":

message = _try_managers("brew", "port")

elif sys.platform.startswith("linux"):

release = platform.linux_distribution()[0].lower()

if release in ('debian', 'ubuntu'):

message = _try_managers('apt-get')

elif release in ('centos', 'redhat', 'fedora'):

message = _try_managers('dnf', 'yum')

return message

class OptionalPackage(SetupPackage):

optional = True

force = False

config_category = "packages"

default_config = "auto"

@classmethod

def get_config(cls):

"""

Look at `setup.cfg` and return one of ["auto", True, False] indicating

if the package is at default state ("auto"), forced by the user (case

insensitively defined as 1, true, yes, on for True) or opted-out (case

insensitively defined as 0, false, no, off for False).

"""

conf = cls.default_config

if config is not None and config.has_option(cls.config_category, cls.name):

try:

conf = config.getboolean(cls.config_category, cls.name)

except ValueError:

conf = config.get(cls.config_category, cls.name)

return conf

def check(self):

"""

Do not override this method!

For custom dependency checks override self.check_requirements().

Two things are checked: Configuration file and requirements.

"""

# Check configuration file

conf = self.get_config()

# Default "auto" state or install forced by user

if conf in [True, 'auto']:

message = "installing"

# Set non-optional if user sets `True` in config

if conf is True:

self.optional = False

# Configuration opt-out by user

else:

# Some backend extensions (e.g. Agg) need to be built for certain

# other GUI backends (e.g. TkAgg) even when manually disabled

if self.force is True:

message = "installing forced (config override)"

else:

raise CheckFailed("skipping due to configuration")

# Check requirements and add extra information (if any) to message.

# If requirements are not met a CheckFailed should be raised in there.

additional_info = self.check_requirements()

if additional_info:

message += ", " + additional_info

# No CheckFailed raised until now, return install message.

return message

def check_requirements(self):

"""

Override this method to do custom dependency checks.

- Raise CheckFailed() if requirements are not met.

- Return message with additional information, or an empty string

(or None) for no additional information.

"""

return ""

class OptionalBackendPackage(OptionalPackage):

config_category = "gui_support"

class Platform(SetupPackage):

name = "platform"

def check(self):

return sys.platform

class Python(SetupPackage):

name = "python"

def check(self):

major, minor1, minor2, s, tmp = sys.version_info

if major < 2:

raise CheckFailed(

"Requires Python 2.7 or later")

elif major == 2 and minor1 < 7:

raise CheckFailed(

"Requires Python 2.7 or later (in the 2.x series)")

elif major == 3 and minor1 < 4:

raise CheckFailed(

"Requires Python 3.4 or later (in the 3.x series)")

return sys.version

class Matplotlib(SetupPackage):

name = "matplotlib"

def check(self):

return versioneer.get_version()

def get_packages(self):

return [

'matplotlib',

'matplotlib.backends',

'matplotlib.backends.qt_editor',

'matplotlib.compat',

'matplotlib.projections',

'matplotlib.axes',

'matplotlib.sphinxext',

'matplotlib.style',

'matplotlib.testing',

'matplotlib.testing._nose',

'matplotlib.testing._nose.plugins',

'matplotlib.testing.jpl_units',

'matplotlib.tri',

'matplotlib.cbook'

]

def get_py_modules(self):

return ['pylab']

def get_package_data(self):

return {

'matplotlib':

[

'mpl-data/fonts/afm/*.afm',

'mpl-data/fonts/pdfcorefonts/*.afm',

'mpl-data/fonts/pdfcorefonts/*.txt',

'mpl-data/fonts/ttf/*.ttf',

'mpl-data/fonts/ttf/LICENSE_STIX',

'mpl-data/fonts/ttf/COPYRIGHT.TXT',

'mpl-data/fonts/ttf/README.TXT',

'mpl-data/fonts/ttf/RELEASENOTES.TXT',

'mpl-data/images/*.xpm',

'mpl-data/images/*.svg',

'mpl-data/images/*.gif',

'mpl-data/images/*.pdf',

'mpl-data/images/*.png',

'mpl-data/images/*.ppm',

'mpl-data/example/*.npy',

'mpl-data/matplotlibrc',

'backends/web_backend/*.*',

'backends/web_backend/js/*.*',

'backends/web_backend/jquery/js/*.min.js',

'backends/web_backend/jquery/css/themes/base/*.min.css',

'backends/web_backend/jquery/css/themes/base/images/*',

'backends/web_backend/css/*.*',

'backends/Matplotlib.nib/*',

'mpl-data/stylelib/*.mplstyle',

]}

class SampleData(OptionalPackage):

"""

This handles the sample data that ships with matplotlib. It is

technically optional, though most often will be desired.

"""

name = "sample_data"

def get_package_data(self):

return {

'matplotlib':

[

'mpl-data/sample_data/*.*',

'mpl-data/sample_data/axes_grid/*.*',

]}

class Toolkits(OptionalPackage):

name = "toolkits"

def get_packages(self):

return [

'mpl_toolkits',

'mpl_toolkits.mplot3d',

'mpl_toolkits.axes_grid',

'mpl_toolkits.axes_grid1',

'mpl_toolkits.axisartist',

]

def get_namespace_packages(self):

return ['mpl_toolkits']

class Tests(OptionalPackage):

name = "tests"

pytest_min_version = '3.6'

default_config = False

def check(self):

super(Tests, self).check()

msgs = []

msg_template = ('{package} is required to run the Matplotlib test '

'suite. Please install it with pip or your preferred '

'tool to run the test suite')

bad_pytest = msg_template.format(

package='pytest %s or later' % self.pytest_min_version

)

try:

import pytest

if is_min_version(pytest.__version__, self.pytest_min_version):

msgs += ['using pytest version %s' % pytest.__version__]

else:

msgs += [bad_pytest]

except ImportError:

msgs += [bad_pytest]

if PY3min:

msgs += ['using unittest.mock']

else:

try:

import mock

msgs += ['using mock %s' % mock.__version__]

except ImportError:

msgs += [msg_template.format(package='mock')]

return ' / '.join(msgs)

def get_packages(self):

return [

'matplotlib.tests',

'matplotlib.sphinxext.tests',

]

def get_package_data(self):

baseline_images = [

'tests/baseline_images/%s/*' % x

for x in os.listdir('lib/matplotlib/tests/baseline_images')]

return {

'matplotlib':

baseline_images +

[

'tests/cmr10.pfb',

'tests/mpltest.ttf',

'tests/test_rcparams.rc',

'tests/test_utf32_be_rcparams.rc',

'sphinxext/tests/tinypages/*.rst',

'sphinxext/tests/tinypages/*.py',

'sphinxext/tests/tinypages/_static/*',

]}

class Toolkits_Tests(Tests):

name = "toolkits_tests"

def check_requirements(self):

conf = self.get_config()

toolkits_conf = Toolkits.get_config()

tests_conf = Tests.get_config()

if conf is True:

Tests.force = True

Toolkits.force = True

elif conf == "auto" and not (toolkits_conf and tests_conf):

# Only auto-install if both toolkits and tests are set

# to be installed

raise CheckFailed("toolkits_tests needs 'toolkits' and 'tests'")

return ""

def get_packages(self):

return [

'mpl_toolkits.tests',

]

def get_package_data(self):

baseline_images = [

'tests/baseline_images/%s/*' % x

for x in os.listdir('lib/mpl_toolkits/tests/baseline_images')]

return {'mpl_toolkits': baseline_images}