|
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