Made test more versatile

Impaler343 · Impaler343 · commit 87066ba142a3 · 2024-05-19T14:25:03.000+05:30
diff --git a/lib/matplotlib/tests/test_pyplot.py b/lib/matplotlib/tests/test_pyplot.py
@@ -460,6 +460,9 @@ def test_figure_hook():
 
 
 def test_multiple_same_figure_calls():
+    fig = mpl.pyplot.figure(1, figsize=(1, 2))
     with pytest.warns(UserWarning, match="Ignoring specified arguments in this call"):
-        fig = mpl.pyplot.figure(1, figsize=(1, 2))
-        fig = mpl.pyplot.figure(1, figsize=(3, 4))
+        fig2 = mpl.pyplot.figure(1, figsize=(3, 4))
+    assert fig is fig2
+    fig3 = mpl.pyplot.figure(1)  # Checks for false warnings
+    assert fig is fig3
