|
1 | | -import datetime |
| 1 | +import datetime |
2 | 2 | import numpy as np |
3 | 3 | from matplotlib.testing.decorators import image_comparison |
4 | 4 | import matplotlib.pyplot as plt |
5 | 5 |
|
6 | | -@image_comparison(baseline_images=['date_empty.png']) |
| 6 | +@image_comparison(baseline_images=['date_empty']) |
7 | 7 | def test_date_empty(): |
8 | 8 | # make sure mpl does the right thing when told to plot dates even |
9 | 9 | # if no date data has been presented, cf |
10 | 10 | # http://sourceforge.net/tracker/?func=detail&aid=2850075&group_id=80706&atid=560720 |
11 | 11 | fig = plt.figure() |
12 | 12 | ax = fig.add_subplot(1,1,1) |
13 | 13 | ax.xaxis_date() |
14 | | - fig.savefig('date_empty.png') |
| 14 | + fig.savefig('date_empty') |
15 | 15 |
|
16 | | -@image_comparison(baseline_images=['date_axhspan.png']) |
| 16 | +@image_comparison(baseline_images=['date_axhspan']) |
17 | 17 | def test_date_axhspan(): |
18 | 18 | # test ax hspan with date inputs |
19 | 19 | t0 = datetime.datetime(2009, 1, 20) |
20 | 20 | tf = datetime.datetime(2009, 1, 21) |
21 | 21 | fig = plt.figure() |
22 | 22 | ax = fig.add_subplot(1,1,1) |
23 | 23 | ax.axhspan( t0, tf, facecolor="blue", alpha=0.25 ) |
24 | | - ax.set_ylim(t0-datetime.timedelta(days=5), |
| 24 | + ax.set_ylim(t0-datetime.timedelta(days=5), |
25 | 25 | tf+datetime.timedelta(days=5)) |
26 | 26 | fig.subplots_adjust(left=0.25) |
27 | | - fig.savefig('date_axhspan.png') |
| 27 | + fig.savefig('date_axhspan') |
28 | 28 |
|
29 | | -@image_comparison(baseline_images=['date_axvspan.png']) |
| 29 | +@image_comparison(baseline_images=['date_axvspan']) |
30 | 30 | def test_date_axvspan(): |
31 | 31 | # test ax hspan with date inputs |
32 | 32 | t0 = datetime.datetime(2000, 1, 20) |
33 | 33 | tf = datetime.datetime(2010, 1, 21) |
34 | 34 | fig = plt.figure() |
35 | 35 | ax = fig.add_subplot(1,1,1) |
36 | 36 | ax.axvspan( t0, tf, facecolor="blue", alpha=0.25 ) |
37 | | - ax.set_xlim(t0-datetime.timedelta(days=720), |
| 37 | + ax.set_xlim(t0-datetime.timedelta(days=720), |
38 | 38 | tf+datetime.timedelta(days=720)) |
39 | 39 | fig.autofmt_xdate() |
40 | | - fig.savefig('date_axvspan.png') |
| 40 | + fig.savefig('date_axvspan') |
41 | 41 |
|
42 | 42 |
|
43 | | -@image_comparison(baseline_images=['date_axhline.png']) |
| 43 | +@image_comparison(baseline_images=['date_axhline']) |
44 | 44 | def test_date_axhline(): |
45 | 45 | # test ax hline with date inputs |
46 | 46 | t0 = datetime.datetime(2009, 1, 20) |
47 | 47 | tf = datetime.datetime(2009, 1, 31) |
48 | 48 | fig = plt.figure() |
49 | 49 | ax = fig.add_subplot(1,1,1) |
50 | 50 | ax.axhline( t0, color="blue", lw=3) |
51 | | - ax.set_ylim(t0-datetime.timedelta(days=5), |
| 51 | + ax.set_ylim(t0-datetime.timedelta(days=5), |
52 | 52 | tf+datetime.timedelta(days=5)) |
53 | 53 | fig.subplots_adjust(left=0.25) |
54 | | - fig.savefig('date_axhline.png') |
| 54 | + fig.savefig('date_axhline') |
55 | 55 |
|
56 | | -@image_comparison(baseline_images=['date_axvline.png']) |
| 56 | +@image_comparison(baseline_images=['date_axvline']) |
57 | 57 | def test_date_axvline(): |
58 | 58 | # test ax hline with date inputs |
59 | 59 | t0 = datetime.datetime(2000, 1, 20) |
60 | 60 | tf = datetime.datetime(2000, 1, 21) |
61 | 61 | fig = plt.figure() |
62 | 62 | ax = fig.add_subplot(1,1,1) |
63 | 63 | ax.axvline( t0, color="red", lw=3) |
64 | | - ax.set_xlim(t0-datetime.timedelta(days=5), |
| 64 | + ax.set_xlim(t0-datetime.timedelta(days=5), |
65 | 65 | tf+datetime.timedelta(days=5)) |
66 | 66 | fig.autofmt_xdate() |
67 | | - fig.savefig('date_axvline.png') |
| 67 | + fig.savefig('date_axvline') |
68 | 68 |
|
69 | 69 |
|
70 | 70 | if __name__=='__main__': |
71 | 71 | import nose |
72 | | - nose.runmodule(argv=['-s','--with-doctest'], exit=False) |
73 | | - |
74 | | - |
75 | | - |
| 72 | + nose.runmodule(argv=['-s','--with-doctest'], exit=False) |
0 commit comments