79 lines
1.9 KiB
Python
79 lines
1.9 KiB
Python
import numpy as np
|
|
from numpy.testing import assert_array_equal
|
|
import matplotlib.pyplot as plt
|
|
|
|
|
|
def test_stem_remove():
|
|
ax = plt.gca()
|
|
st = ax.stem([1, 2], [1, 2])
|
|
st.remove()
|
|
|
|
|
|
def test_errorbar_remove():
|
|
|
|
# Regression test for a bug that caused remove to fail when using
|
|
# fmt='none'
|
|
|
|
ax = plt.gca()
|
|
|
|
eb = ax.errorbar([1], [1])
|
|
eb.remove()
|
|
|
|
eb = ax.errorbar([1], [1], xerr=1)
|
|
eb.remove()
|
|
|
|
eb = ax.errorbar([1], [1], yerr=2)
|
|
eb.remove()
|
|
|
|
eb = ax.errorbar([1], [1], xerr=[2], yerr=2)
|
|
eb.remove()
|
|
|
|
eb = ax.errorbar([1], [1], fmt='none')
|
|
eb.remove()
|
|
|
|
|
|
def test_nonstring_label():
|
|
# Test for #26824
|
|
plt.bar(np.arange(10), np.random.rand(10), label=1)
|
|
plt.legend()
|
|
|
|
|
|
def test_barcontainer_position_centers__bottoms__tops():
|
|
fig, ax = plt.subplots()
|
|
pos = [1, 2, 4]
|
|
bottoms = np.array([1, 5, 3])
|
|
heights = np.array([2, 3, 4])
|
|
|
|
container = ax.bar(pos, heights, bottom=bottoms)
|
|
assert_array_equal(container.position_centers, pos)
|
|
assert_array_equal(container.bottoms, bottoms)
|
|
assert_array_equal(container.tops, bottoms + heights)
|
|
|
|
container = ax.barh(pos, heights, left=bottoms)
|
|
assert_array_equal(container.position_centers, pos)
|
|
assert_array_equal(container.bottoms, bottoms)
|
|
assert_array_equal(container.tops, bottoms + heights)
|
|
|
|
|
|
def test_piecontainer_remove():
|
|
fig, ax = plt.subplots()
|
|
pie = ax.pie([2, 3], labels=['foo', 'bar'], autopct="%1.0f%%")
|
|
ax.pie_label(pie, ['baz', 'qux'])
|
|
assert len(ax.patches) == 2
|
|
assert len(ax.texts) == 6
|
|
|
|
pie.remove()
|
|
assert not ax.patches
|
|
assert not ax.texts
|
|
|
|
|
|
def test_piecontainer_unpack_backcompat():
|
|
fig, ax = plt.subplots()
|
|
wedges, texts, autotexts = ax.pie(
|
|
[2, 3], labels=['foo', 'bar'], autopct="%1.0f%%", labeldistance=None)
|
|
|
|
assert len(wedges) == 2
|
|
assert isinstance(texts, list)
|
|
assert not texts
|
|
assert len(autotexts) == 2
|