from __future__ import (absolute_import, division, print_function,

unicode_literals)

import six

import os

import sys

import warnings

import matplotlib as mpl

from matplotlib.tests import assert_str_equal

from nose.tools import assert_true, assert_raises

import nose

mpl.rc('text', usetex=False)

mpl.rc('lines', linewidth=22)

fname = os.path.join(os.path.dirname(__file__), 'test_rcparams.rc')

def test_rcparams():

usetex = mpl.rcParams['text.usetex']

linewidth = mpl.rcParams['lines.linewidth']

# test context given dictionary

with mpl.rc_context(rc={'text.usetex': not usetex}):

assert mpl.rcParams['text.usetex'] == (not usetex)

assert mpl.rcParams['text.usetex'] == usetex

# test context given filename (mpl.rc sets linewdith to 33)

with mpl.rc_context(fname=fname):

assert mpl.rcParams['lines.linewidth'] == 33

assert mpl.rcParams['lines.linewidth'] == linewidth

# test context given filename and dictionary

with mpl.rc_context(fname=fname, rc={'lines.linewidth': 44}):

assert mpl.rcParams['lines.linewidth'] == 44

assert mpl.rcParams['lines.linewidth'] == linewidth

# test rc_file

try:

mpl.rc_file(fname)

assert mpl.rcParams['lines.linewidth'] == 33

finally:

mpl.rcParams['lines.linewidth'] = linewidth

def test_RcParams_class():

rc = mpl.RcParams({'font.cursive': ['Apple Chancery',

'Textile',

'Zapf Chancery',

'cursive'],

'font.family': 'sans-serif',

'font.weight': 'normal',

'font.size': 12})

if six.PY3:

expected_repr = """

RcParams({'font.cursive': ['Apple Chancery',

'Textile',

'Zapf Chancery',

'cursive'],

'font.family': 'sans-serif',

'font.size': 12,

'font.weight': 'normal'})""".lstrip()

else:

expected_repr = """

RcParams({u'font.cursive': [u'Apple Chancery',

u'Textile',

u'Zapf Chancery',

u'cursive'],

u'font.family': u'sans-serif',

u'font.size': 12,

u'font.weight': u'normal'})""".lstrip()

assert_str_equal(expected_repr, repr(rc))

if six.PY3:

expected_str = """

font.cursive: ['Apple Chancery', 'Textile', 'Zapf Chancery', 'cursive']

font.family: sans-serif

font.size: 12

font.weight: normal""".lstrip()

else:

expected_str = """

font.cursive: [u'Apple Chancery', u'Textile', u'Zapf Chancery', u'cursive']

font.family: sans-serif

font.size: 12

font.weight: normal""".lstrip()

assert_str_equal(expected_str, str(rc))

# test the find_all functionality

assert ['font.cursive', 'font.size'] == sorted(rc.find_all('i[vz]').keys())

assert ['font.family'] == list(six.iterkeys(rc.find_all('family')))

def test_Bug_2543():

# Test that it possible to add all values to itself / deepcopy

# This was not possible because validate_bool_maybe_none did not

# accept None as an argument.

# https://github.com/matplotlib/matplotlib/issues/2543

# We filter warnings at this stage since a number of them are raised

# for deprecated rcparams as they should. We dont want these in the

# printed in the test suite.

with warnings.catch_warnings():

warnings.filterwarnings('ignore',

message='.*(deprecated|obsolete)',

category=UserWarning)

with mpl.rc_context():

_copy = mpl.rcParams.copy()

for key in six.iterkeys(_copy):

mpl.rcParams[key] = _copy[key]

mpl.rcParams['text.dvipnghack'] = None

with mpl.rc_context():

from copy import deepcopy

_deep_copy = deepcopy(mpl.rcParams)

from matplotlib.rcsetup import validate_bool_maybe_none, validate_bool

# real test is that this does not raise

assert_true(validate_bool_maybe_none(None) is None)

assert_true(validate_bool_maybe_none("none") is None)

_fonttype = mpl.rcParams['svg.fonttype']

assert_true(_fonttype == mpl.rcParams['svg.embed_char_paths'])

with mpl.rc_context():

mpl.rcParams['svg.embed_char_paths'] = False

assert_true(mpl.rcParams['svg.fonttype'] == "none")

def test_Bug_2543_newer_python():

# only split from above because of the usage of assert_raises

# as a context manager, which only works in 2.7 and above

if sys.version_info[:2] < (2, 7):

raise nose.SkipTest("assert_raises as context manager not supported with Python < 2.7")

from matplotlib.rcsetup import validate_bool_maybe_none, validate_bool

with assert_raises(ValueError):

validate_bool_maybe_none("blah")

with assert_raises(ValueError):

validate_bool(None)

with assert_raises(ValueError):

with mpl.rc_context():

mpl.rcParams['svg.fonttype'] = True

if __name__ == '__main__':

import nose

nose.runmodule(argv=['-s', '--with-doctest'], exit=False)