|
8 | 8 |
|
9 | 9 | import numpy as np
|
10 | 10 | from numpy.testing import assert_array_equal
|
| 11 | +from numpy.testing import assert_equal |
11 | 12 | from numpy.testing import assert_almost_equal
|
12 | 13 |
|
13 | 14 | from matplotlib.patches import Polygon
|
@@ -202,6 +203,21 @@ def test_patch_custom_linestyle():
|
202 | 203 | ax.set_ylim([-1, 2])
|
203 | 204 |
|
204 | 205 |
|
| 206 | +def test_wedge_movement(): |
| 207 | + param_dict = {'center': ((0, 0), (1, 1), 'set_center'), |
| 208 | + 'r': (5, 8, 'set_radius'), |
| 209 | + 'width': (2, 3, 'set_width'), |
| 210 | + 'theta1': (0, 30, 'set_theta1'), |
| 211 | + 'theta2': (45, 50, 'set_theta2')} |
| 212 | + |
| 213 | + init_args = dict((k, v[0]) for (k, v) in six.iteritems(param_dict)) |
| 214 | + |
| 215 | + w = mpatches.Wedge(**init_args) |
| 216 | + for attr, (old_v, new_v, func) in six.iteritems(param_dict): |
| 217 | + assert_equal(getattr(w, attr), old_v) |
| 218 | + getattr(w, func)(new_v) |
| 219 | + assert_equal(getattr(w, attr), new_v) |
| 220 | + |
205 | 221 | if __name__ == '__main__':
|
206 | 222 | import nose
|
207 | 223 | nose.runmodule(argv=['-s', '--with-doctest'], exit=False)
|
0 commit comments