171 lines
5.7 KiB
Python
171 lines
5.7 KiB
Python
import contextlib
|
|
from io import StringIO
|
|
|
|
import matplotlib as mpl
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
import pytest
|
|
|
|
from cycler import cycler
|
|
|
|
|
|
def test_colorcycle_basic():
|
|
fig, ax = plt.subplots()
|
|
ax.set_prop_cycle(cycler('color', ['r', 'g', 'y']))
|
|
for _ in range(4):
|
|
ax.plot(range(10), range(10))
|
|
assert [l.get_color() for l in ax.lines] == ['r', 'g', 'y', 'r']
|
|
|
|
|
|
def test_marker_cycle():
|
|
fig, ax = plt.subplots()
|
|
ax.set_prop_cycle(cycler('c', ['r', 'g', 'y']) +
|
|
cycler('marker', ['.', '*', 'x']))
|
|
for _ in range(4):
|
|
ax.plot(range(10), range(10))
|
|
assert [l.get_color() for l in ax.lines] == ['r', 'g', 'y', 'r']
|
|
assert [l.get_marker() for l in ax.lines] == ['.', '*', 'x', '.']
|
|
|
|
|
|
def test_marker_cycle_kwargs_arrays_iterators():
|
|
fig, ax = plt.subplots()
|
|
ax.set_prop_cycle(c=np.array(['r', 'g', 'y']),
|
|
marker=iter(['.', '*', 'x']))
|
|
for _ in range(4):
|
|
ax.plot(range(10), range(10))
|
|
assert [l.get_color() for l in ax.lines] == ['r', 'g', 'y', 'r']
|
|
assert [l.get_marker() for l in ax.lines] == ['.', '*', 'x', '.']
|
|
|
|
|
|
def test_linestylecycle_basic():
|
|
fig, ax = plt.subplots()
|
|
ax.set_prop_cycle(cycler('ls', ['-', '--', ':']))
|
|
for _ in range(4):
|
|
ax.plot(range(10), range(10))
|
|
assert [l.get_linestyle() for l in ax.lines] == ['-', '--', ':', '-']
|
|
|
|
|
|
def test_fillcycle_basic():
|
|
fig, ax = plt.subplots()
|
|
ax.set_prop_cycle(cycler('c', ['r', 'g', 'y']) +
|
|
cycler('hatch', ['xx', 'O', '|-']) +
|
|
cycler('linestyle', ['-', '--', ':']))
|
|
for _ in range(4):
|
|
ax.fill(range(10), range(10))
|
|
assert ([p.get_facecolor() for p in ax.patches]
|
|
== [mpl.colors.to_rgba(c) for c in ['r', 'g', 'y', 'r']])
|
|
assert [p.get_hatch() for p in ax.patches] == ['xx', 'O', '|-', 'xx']
|
|
assert [p.get_linestyle() for p in ax.patches] == ['-', '--', ':', '-']
|
|
|
|
|
|
def test_fillcycle_ignore():
|
|
fig, ax = plt.subplots()
|
|
ax.set_prop_cycle(cycler('color', ['r', 'g', 'y']) +
|
|
cycler('hatch', ['xx', 'O', '|-']) +
|
|
cycler('marker', ['.', '*', 'D']))
|
|
t = range(10)
|
|
# Should not advance the cycler, even though there is an
|
|
# unspecified property in the cycler "marker".
|
|
# "marker" is not a Polygon property, and should be ignored.
|
|
ax.fill(t, t, 'r', hatch='xx')
|
|
# Allow the cycler to advance, but specify some properties
|
|
ax.fill(t, t, hatch='O')
|
|
ax.fill(t, t)
|
|
ax.fill(t, t)
|
|
assert ([p.get_facecolor() for p in ax.patches]
|
|
== [mpl.colors.to_rgba(c) for c in ['r', 'r', 'g', 'y']])
|
|
assert [p.get_hatch() for p in ax.patches] == ['xx', 'O', 'O', '|-']
|
|
|
|
|
|
def test_property_collision_plot():
|
|
fig, ax = plt.subplots()
|
|
ax.set_prop_cycle('linewidth', [2, 4])
|
|
t = range(10)
|
|
for c in range(1, 4):
|
|
ax.plot(t, t, lw=0.1)
|
|
ax.plot(t, t)
|
|
ax.plot(t, t)
|
|
assert [l.get_linewidth() for l in ax.lines] == [0.1, 0.1, 0.1, 2, 4]
|
|
|
|
|
|
def test_property_collision_fill():
|
|
fig, ax = plt.subplots()
|
|
ax.set_prop_cycle(linewidth=[2, 3, 4, 5, 6], facecolor='bgcmy')
|
|
t = range(10)
|
|
for c in range(1, 4):
|
|
ax.fill(t, t, lw=0.1)
|
|
ax.fill(t, t)
|
|
ax.fill(t, t)
|
|
assert ([p.get_facecolor() for p in ax.patches]
|
|
== [mpl.colors.to_rgba(c) for c in 'bgcmy'])
|
|
assert [p.get_linewidth() for p in ax.patches] == [0.1, 0.1, 0.1, 5, 6]
|
|
|
|
|
|
def test_valid_input_forms():
|
|
fig, ax = plt.subplots()
|
|
# These should not raise an error.
|
|
ax.set_prop_cycle(None)
|
|
ax.set_prop_cycle(cycler('linewidth', [1, 2]))
|
|
ax.set_prop_cycle('color', 'rgywkbcm')
|
|
ax.set_prop_cycle('lw', (1, 2))
|
|
ax.set_prop_cycle('linewidth', [1, 2])
|
|
ax.set_prop_cycle('linewidth', iter([1, 2]))
|
|
ax.set_prop_cycle('linewidth', np.array([1, 2]))
|
|
ax.set_prop_cycle('color', np.array([[1, 0, 0],
|
|
[0, 1, 0],
|
|
[0, 0, 1]]))
|
|
ax.set_prop_cycle('dashes', [[], [13, 2], [8, 3, 1, 3]])
|
|
ax.set_prop_cycle(lw=[1, 2], color=['k', 'w'], ls=['-', '--'])
|
|
ax.set_prop_cycle(lw=np.array([1, 2]),
|
|
color=np.array(['k', 'w']),
|
|
ls=np.array(['-', '--']))
|
|
|
|
|
|
def test_cycle_reset():
|
|
fig, ax = plt.subplots()
|
|
prop0 = StringIO()
|
|
prop1 = StringIO()
|
|
prop2 = StringIO()
|
|
|
|
with contextlib.redirect_stdout(prop0):
|
|
plt.getp(ax.plot([1, 2], label="label")[0])
|
|
|
|
ax.set_prop_cycle(linewidth=[10, 9, 4])
|
|
with contextlib.redirect_stdout(prop1):
|
|
plt.getp(ax.plot([1, 2], label="label")[0])
|
|
assert prop1.getvalue() != prop0.getvalue()
|
|
|
|
ax.set_prop_cycle(None)
|
|
with contextlib.redirect_stdout(prop2):
|
|
plt.getp(ax.plot([1, 2], label="label")[0])
|
|
assert prop2.getvalue() == prop0.getvalue()
|
|
|
|
|
|
def test_invalid_input_forms():
|
|
fig, ax = plt.subplots()
|
|
|
|
with pytest.raises((TypeError, ValueError)):
|
|
ax.set_prop_cycle(1)
|
|
with pytest.raises((TypeError, ValueError)):
|
|
ax.set_prop_cycle([1, 2])
|
|
|
|
with pytest.raises((TypeError, ValueError)):
|
|
ax.set_prop_cycle('color', 'fish')
|
|
|
|
with pytest.raises((TypeError, ValueError)):
|
|
ax.set_prop_cycle('linewidth', 1)
|
|
with pytest.raises((TypeError, ValueError)):
|
|
ax.set_prop_cycle('linewidth', {1, 2})
|
|
with pytest.raises((TypeError, ValueError)):
|
|
ax.set_prop_cycle(linewidth=1, color='r')
|
|
|
|
with pytest.raises((TypeError, ValueError)):
|
|
ax.set_prop_cycle('foobar', [1, 2])
|
|
with pytest.raises((TypeError, ValueError)):
|
|
ax.set_prop_cycle(foobar=[1, 2])
|
|
|
|
with pytest.raises((TypeError, ValueError)):
|
|
ax.set_prop_cycle(cycler(foobar=[1, 2]))
|
|
with pytest.raises(ValueError):
|
|
ax.set_prop_cycle(cycler(color='rgb', c='cmy'))
|