File tree Expand file tree Collapse file tree 2 files changed +20
-1
lines changed Expand file tree Collapse file tree 2 files changed +20
-1
lines changed Original file line number Diff line number Diff line change @@ -460,7 +460,9 @@ def gcf():
460460 else :
461461 return figure ()
462462
463- fignum_exists = _pylab_helpers .Gcf .has_fignum
463+ def fignum_exists (num ):
464+ allLabels = get_figlabels ()
465+ return _pylab_helpers .Gcf .has_fignum (num ) or num in allLabels
464466
465467
466468def get_fignums ():
Original file line number Diff line number Diff line change @@ -32,6 +32,23 @@ def test_figure_label():
3232 assert_equal (plt .get_figlabels (), ['' , 'today' ])
3333
3434
35+ @cleanup
36+ def test_fignum_exists ():
37+ # pyplot figure creation, selection and closing with fignum_exists
38+ plt .figure ('one' )
39+ plt .figure (2 )
40+ plt .figure ('three' )
41+ plt .figure ()
42+ assert_equal (plt .fignum_exists ('one' ), True )
43+ assert_equal (plt .fignum_exists (2 ), True )
44+ assert_equal (plt .fignum_exists ('three' ), True )
45+ assert_equal (plt .fignum_exists (4 ), True )
46+ plt .close ('one' )
47+ plt .close (4 )
48+ assert_equal (plt .fignum_exists ('one' ), False )
49+ assert_equal (plt .fignum_exists (4 ), False )
50+
51+
3552@image_comparison (baseline_images = ['figure_today' ])
3653def test_figure ():
3754 # named figure support
You can’t perform that action at this time.
0 commit comments