|
1 | 1 | import unittest
|
2 | 2 | import numpy as np
|
3 |
| -import matplotlib.axes as axes |
| 3 | +import matplotlib.cbook as cbook |
4 | 4 |
|
5 | 5 | class TestAxes(unittest.TestCase):
|
6 | 6 | def test_delete_masked_points_arrays(self):
|
7 | 7 | input = ( [1,2,3,np.nan,5],
|
8 | 8 | np.array((1,2,3,4,5)),
|
9 | 9 | )
|
10 | 10 | expected = [np.array((1,2,3,5))]*2
|
11 |
| - actual = axes.delete_masked_points(*input) |
| 11 | + actual = cbook.delete_masked_points(*input) |
12 | 12 | assert np.allclose(actual, expected)
|
13 | 13 |
|
14 | 14 | input = ( np.ma.array( [1,2,3,4,5], mask=[False,False,False,True,False] ),
|
15 | 15 | np.array((1,2,3,4,5)),
|
16 | 16 | )
|
17 | 17 | expected = [np.array((1,2,3,5))]*2
|
18 |
| - actual = axes.delete_masked_points(*input) |
| 18 | + actual = cbook.delete_masked_points(*input) |
19 | 19 | assert np.allclose(actual, expected)
|
20 | 20 |
|
21 | 21 | input = ( [1,2,3,np.nan,5],
|
22 | 22 | np.ma.array( [1,2,3,4,5], mask=[False,False,False,True,False] ),
|
23 | 23 | np.array((1,2,3,4,5)),
|
24 | 24 | )
|
25 | 25 | expected = [np.array((1,2,3,5))]*3
|
26 |
| - actual = axes.delete_masked_points(*input) |
| 26 | + actual = cbook.delete_masked_points(*input) |
27 | 27 | assert np.allclose(actual, expected)
|
28 | 28 |
|
29 | 29 | input = ()
|
30 | 30 | expected = ()
|
31 |
| - actual = axes.delete_masked_points(*input) |
| 31 | + actual = cbook.delete_masked_points(*input) |
32 | 32 | assert np.allclose(actual, expected)
|
33 | 33 |
|
34 | 34 |
|
35 | 35 | input = ( [1,2,3,np.nan,5],
|
36 | 36 | )
|
37 | 37 | expected = [np.array((1,2,3,5))]*1
|
38 |
| - actual = axes.delete_masked_points(*input) |
| 38 | + actual = cbook.delete_masked_points(*input) |
39 | 39 | assert np.allclose(actual, expected)
|
40 | 40 |
|
41 | 41 | input = ( np.array((1,2,3,4,5)),
|
42 | 42 | )
|
43 | 43 | expected = [np.array((1,2,3,4,5))]*1
|
44 |
| - actual = axes.delete_masked_points(*input) |
| 44 | + actual = cbook.delete_masked_points(*input) |
45 | 45 | assert np.allclose(actual, expected)
|
46 | 46 |
|
47 | 47 | def test_delete_masked_points_strings(self):
|
48 | 48 | input = ( 'hello',
|
49 | 49 | )
|
50 | 50 | expected = ('hello',)
|
51 |
| - actual = axes.delete_masked_points(*input) |
| 51 | + actual = cbook.delete_masked_points(*input) |
52 | 52 | assert actual == expected
|
53 | 53 |
|
54 | 54 | input = ( u'hello',
|
55 | 55 | )
|
56 | 56 | expected = (u'hello',)
|
57 |
| - actual = axes.delete_masked_points(*input) |
| 57 | + actual = cbook.delete_masked_points(*input) |
58 | 58 | assert actual == expected
|
59 | 59 |
|
60 | 60 |
|
|
0 commit comments