|
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