Add tests

dstansby · dstansby · commit b1b6bea61f7a · 2020-12-28T22:34:51.000Z
diff --git a/lib/matplotlib/tests/test_testing.py b/lib/matplotlib/tests/test_testing.py
@@ -1,5 +1,8 @@
 import warnings
+
 import pytest
+
+import matplotlib.pyplot as plt
 from matplotlib.testing.decorators import check_figures_equal
 
 
@@ -22,3 +25,17 @@ def test_wrap_failure():
         @check_figures_equal()
         def should_fail(test, ref):
             pass
+
+
+@pytest.mark.xfail(raises=RuntimeError, strict=True,
+                   reason='Test for check_figures_equal test creating '
+                          'new figures')
+@check_figures_equal()
+def test_check_figures_equal_extra_fig(fig_test, fig_ref):
+    plt.figure()
+
+
+@check_figures_equal()
+def test_check_figures_equal_closed_fig(fig_test, fig_ref):
+    fig = plt.figure()
+    plt.close(fig)
