File tree Expand file tree Collapse file tree 2 files changed +15
-2
lines changed
Filter options
Expand file tree Collapse file tree 2 files changed +15
-2
lines changed
Original file line number Diff line number Diff line change @@ -29,14 +29,14 @@ def _prevent_rasterization(draw):
29
29
# (e.g., change in dpi).
30
30
31
31
@wraps (draw )
32
- def draw_wrapper (artist , renderer ):
32
+ def draw_wrapper (artist , renderer , * args , ** kwargs ):
33
33
if renderer ._raster_depth == 0 and renderer ._rasterizing :
34
34
# Only stop when we are not in a rasterized parent
35
35
# and something has been rasterized since last stop.
36
36
renderer .stop_rasterizing ()
37
37
renderer ._rasterizing = False
38
38
39
- return draw (artist , renderer )
39
+ return draw (artist , renderer , * args , ** kwargs )
40
40
41
41
draw_wrapper ._supports_rasterization = False
42
42
return draw_wrapper
Original file line number Diff line number Diff line change 13
13
import matplotlib .transforms as mtransforms
14
14
import matplotlib .collections as mcollections
15
15
import matplotlib .artist as martist
16
+ import matplotlib .backend_bases as mbackend_bases
16
17
import matplotlib as mpl
17
18
from matplotlib .testing .decorators import check_figures_equal , image_comparison
18
19
@@ -549,3 +550,15 @@ class Gen2(Gen1):
549
550
550
551
assert 'draw' not in Gen2 .__dict__
551
552
assert Gen2 .draw is Gen1 .draw
553
+
554
+
555
+ def test_draw_wraper_forward_input ():
556
+ class TestKlass (martist .Artist ):
557
+ def draw (self , renderer , extra ):
558
+ return extra
559
+
560
+ art = TestKlass ()
561
+ renderer = mbackend_bases .RendererBase ()
562
+
563
+ assert 'aardvark' == art .draw (renderer , 'aardvark' )
564
+ assert 'aardvark' == art .draw (renderer , extra = 'aardvark' )
You can’t perform that action at this time.
0 commit comments