Skip to content

Commit e1fb9bb

Browse files
committed
new test for overriding color and label kargs
1 parent e687af7 commit e1fb9bb

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed
Loading

probscale/tests/test_viz.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -526,8 +526,6 @@ def test_probplot_test_results(plot_data):
526526
return fig
527527

528528

529-
530-
531529
@pytest.mark.parametrize('probax', ['x', 'y'])
532530
@pytest.mark.parametrize(('N', 'minval', 'maxval'), [
533531
(8, 10, 90),
@@ -544,3 +542,11 @@ def test__set_prob_limits_x(probax, N, minval, maxval):
544542
ax.set_xlim.assert_called_once_with(left=minval, right=maxval)
545543
elif probax == 'y':
546544
ax.set_ylim.assert_called_once_with(bottom=minval, top=maxval)
545+
546+
547+
@pytest.mark.mpl_image_compare(baseline_dir=BASELINE_DIR, tolerance=10)
548+
def test_probplot_color_and_label(plot_data):
549+
fig, ax = plt.subplots()
550+
fig = viz.probplot(plot_data, ax=ax, color='pink', label='A Top-Level Label')
551+
ax.legend(loc='lower right')
552+
return fig

0 commit comments

Comments
 (0)