Skip to content

Commit de62411

Browse files
committed
Add tests for MarkerStyle scaled and translated
1 parent 9a27a92 commit de62411

File tree

1 file changed

+36
-2
lines changed

1 file changed

+36
-2
lines changed

lib/matplotlib/tests/test_marker.py

Lines changed: 36 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -229,7 +229,7 @@ def test_marker_rotated_invalid():
229229
marker = markers.MarkerStyle("o")
230230
with pytest.raises(ValueError):
231231
new_marker = marker.rotated()
232-
with pytest.raises(ValueError):
232+
with pytest.raises(ValueError):
233233
new_marker = marker.rotated(deg=10, rad=10)
234234

235235

@@ -246,8 +246,42 @@ def test_marker_rotated_invalid():
246246
markers.TICKLEFT, transform=Affine2D().translate(1, 1)),
247247
10, None, Affine2D().translate(1, 1).rotate_deg(10)),
248248
])
249-
def test_marker_rotated_deg(marker, deg, rad, expected):
249+
def test_marker_rotated(marker, deg, rad, expected):
250250
new_marker = marker.rotated(deg=deg, rad=rad)
251251
assert new_marker is not marker
252252
assert new_marker.get_user_transform() == expected
253253
assert marker._user_transform is not new_marker._user_transform
254+
255+
256+
def test_marker_translated():
257+
marker = markers.MarkerStyle("1")
258+
new_marker = marker.translated(1, 1)
259+
assert new_marker is not marker
260+
assert new_marker.get_user_transform() == Affine2D().translate(1, 1)
261+
assert marker._user_transform is not new_marker._user_transform
262+
263+
marker = markers.MarkerStyle("1", transform=Affine2D().translate(1, 1))
264+
new_marker = marker.translated(1, 1)
265+
assert new_marker is not marker
266+
assert new_marker.get_user_transform() == Affine2D().translate(2, 2)
267+
assert marker._user_transform is not new_marker._user_transform
268+
269+
270+
def test_marker_scaled():
271+
marker = markers.MarkerStyle("1")
272+
new_marker = marker.scaled(2)
273+
assert new_marker is not marker
274+
assert new_marker.get_user_transform() == Affine2D().scale(2)
275+
assert marker._user_transform is not new_marker._user_transform
276+
277+
new_marker = marker.scaled(2, 3)
278+
assert new_marker is not marker
279+
assert new_marker.get_user_transform() == Affine2D().scale(2, 3)
280+
assert marker._user_transform is not new_marker._user_transform
281+
282+
marker = markers.MarkerStyle("1", transform=Affine2D().translate(1, 1))
283+
new_marker = marker.scaled(2)
284+
assert new_marker is not marker
285+
expected = Affine2D().translate(1, 1).scale(2)
286+
assert new_marker.get_user_transform() == expected
287+
assert marker._user_transform is not new_marker._user_transform

0 commit comments

Comments
 (0)