|
13 | 13 | from matplotlib.legend_handler import HandlerTuple
|
14 | 14 | import matplotlib.legend as mlegend
|
15 | 15 | from matplotlib import rc_context
|
| 16 | +from matplotlib.font_manager import FontProperties |
16 | 17 |
|
17 | 18 |
|
18 | 19 | def test_legend_ordereddict():
|
@@ -546,35 +547,40 @@ def test_window_extent_cached_renderer():
|
546 | 547 |
|
547 | 548 | def test_legend_title_fontprop_fontsize():
|
548 | 549 | # test the title_fontsize kwarg
|
| 550 | + plt.plot(range(10)) |
| 551 | + with pytest.raises(ValueError): |
| 552 | + plt.legend(title='Aardvark', title_fontsize=22, |
| 553 | + title_fontproperties={'family': 'serif', 'size': 22}) |
| 554 | + |
| 555 | + leg = plt.legend(title='Aardvark', title_fontproperties=FontProperties( |
| 556 | + family='serif', size=22)) |
| 557 | + assert leg.get_title().get_size() == 22 |
| 558 | + |
549 | 559 | fig, axes = plt.subplots(2, 3, figsize=(10, 6))
|
550 | 560 | axes = axes.flat
|
551 | 561 | axes[0].plot(range(10))
|
552 |
| - leg = axes[0].legend(title='Aardvark', title_fontsize=22) |
553 |
| - assert leg.get_title().get_fontsize() == 22 |
| 562 | + leg0 = axes[0].legend(title='Aardvark', title_fontsize=22) |
| 563 | + assert leg0.get_title().get_fontsize() == 22 |
554 | 564 | axes[1].plot(range(10))
|
555 |
| - leg2 = axes[1].legend(title='Aardvark', |
| 565 | + leg1 = axes[1].legend(title='Aardvark', |
556 | 566 | title_fontproperties={'family': 'serif', 'size': 22})
|
557 |
| - assert leg2.get_title().get_fontsize() == 22 |
| 567 | + assert leg1.get_title().get_fontsize() == 22 |
558 | 568 | axes[2].plot(range(10))
|
559 |
| - leg3 = axes[2].legend(title='Aardvark', |
560 |
| - title_fontproperties={'family': 'serif'}, |
561 |
| - title_fontsize=22) |
562 |
| - assert leg3.get_title().get_fontsize() == 22 |
563 |
| - axes[3].plot(range(10)) |
564 |
| - leg4 = axes[3].legend(title='Aardvark', |
565 |
| - title_fontproperties={'family': 'serif', 'size': 10}, |
566 |
| - title_fontsize=22) |
567 |
| - assert leg4.get_title().get_fontsize() == 22 |
568 | 569 | mpl.rcParams['legend.title_fontsize'] = None
|
569 |
| - axes[4].plot(range(10)) |
570 |
| - leg5 = axes[4].legend(title='Aardvark', |
| 570 | + leg2 = axes[2].legend(title='Aardvark', |
571 | 571 | title_fontproperties={'family': 'serif'})
|
572 |
| - assert leg5.get_title().get_fontsize() == mpl.rcParams['font.size'] |
| 572 | + assert leg2.get_title().get_fontsize() == mpl.rcParams['font.size'] |
| 573 | + axes[3].plot(range(10)) |
| 574 | + leg3 = axes[3].legend(title='Aardvark') |
| 575 | + assert leg3.get_title().get_fontsize() == mpl.rcParams['font.size'] |
| 576 | + axes[4].plot(range(10)) |
573 | 577 | mpl.rcParams['legend.title_fontsize'] = 20
|
574 |
| - axes[5].plot(range(10)) |
575 |
| - leg6 = axes[5].legend(title='Aardvark', |
| 578 | + leg4 = axes[4].legend(title='Aardvark', |
576 | 579 | title_fontproperties={'family': 'serif'})
|
577 |
| - assert leg6.get_title().get_fontsize() == 20 |
| 580 | + assert leg4.get_title().get_fontsize() == 20 |
| 581 | + axes[5].plot(range(10)) |
| 582 | + leg5 = axes[5].legend(title='Aardvark') |
| 583 | + assert leg5.get_title().get_fontsize() == 20 |
578 | 584 |
|
579 | 585 |
|
580 | 586 | def test_legend_labelcolor_single():
|
|
0 commit comments