TST: simplify tests + comments

tacaswell · QuLogic · web-flow · commit 00829a661cfd · 2022-03-10T10:18:06.000-05:00
Co-authored-by: Elliott Sales de Andrade &lt;quantum.analyst@gmail.com&gt;
diff --git a/lib/matplotlib/tests/test_figure.py b/lib/matplotlib/tests/test_figure.py
@@ -740,11 +740,11 @@ def test_figure_clear():
     subaxes = subfig.add_subplot(111)
     mainaxes = fig.add_subplot(gs[1])
 
-    # e.1) removing just the axis leaves the subplot
+    # e.1) removing just the axes leaves the subplot
     mainaxes.remove()
     assert [subaxes] == fig.axes
 
-    # e.2) removing just the subaxis leaves the subplot
+    # e.2) removing just the subaxes leaves the subplot
     # and subfigure
     mainaxes = fig.add_subplot(gs[1])
     subaxes.remove()
@@ -753,7 +753,8 @@ def test_figure_clear():
 
     # e.3) clearing the subfigure leaves the subplot
     subaxes = subfig.add_subplot(111)
-    assert all(ax in fig.axes for ax in [mainaxes, subaxes])
+    assert mainaxes in fig.axes
+    assert subaxes in fig.axes
     subfig.clear()
     assert subfig in fig.subfigs
     assert subaxes not in subfig.axes
