Update tests for data containers

ksunden · ksunden · commit 1c0c88473886 · 2025-12-04T12:58:45.000-06:00
diff --git a/lib/matplotlib/tests/test_collections.py b/lib/matplotlib/tests/test_collections.py
@@ -837,16 +837,16 @@ def test_collection_set_verts_array():
     verts = np.arange(80, dtype=np.double).reshape(10, 4, 2)
     col_arr = PolyCollection(verts)
     col_list = PolyCollection(list(verts))
-    assert len(col_arr._paths) == len(col_list._paths)
-    for ap, lp in zip(col_arr._paths, col_list._paths):
+    assert len(col_arr.get_paths()) == len(col_list.get_paths())
+    for ap, lp in zip(col_arr.get_paths(), col_list.get_paths()):
         assert np.array_equal(ap._vertices, lp._vertices)
         assert np.array_equal(ap._codes, lp._codes)
 
     verts_tuple = np.empty(10, dtype=object)
     verts_tuple[:] = [tuple(tuple(y) for y in x) for x in verts]
     col_arr_tuple = PolyCollection(verts_tuple)
-    assert len(col_arr._paths) == len(col_arr_tuple._paths)
-    for ap, atp in zip(col_arr._paths, col_arr_tuple._paths):
+    assert len(col_arr.get_paths()) == len(col_arr_tuple.get_paths())
+    for ap, atp in zip(col_arr.get_paths(), col_arr_tuple.get_paths()):
         assert np.array_equal(ap._vertices, atp._vertices)
         assert np.array_equal(ap._codes, atp._codes)
 
diff --git a/lib/matplotlib/tests/test_image.py b/lib/matplotlib/tests/test_image.py
@@ -781,7 +781,9 @@ def test_setdata_xya(image_cls, x, y, a):
     im = image_cls(ax)
     im.set_data(x, y, a)
     x[0] = y[0] = a[0, 0] = 9.9
-    assert im._A[0, 0] == im._Ax[0] == im._Ay[0] == 0, 'value changed'
+    Ax = im._container.x
+    Ay = im._container.y
+    assert im._A[0, 0] == Ax[0] == Ay[0] == 0, 'value changed'
     im.set_data(x, y, a.reshape((*a.shape, -1)))  # Just a smoketest.
 
 
@@ -1660,8 +1662,8 @@ def test_axesimage_get_shape():
     # generate dummy image to test get_shape method
     ax = plt.gca()
     im = AxesImage(ax)
-    with pytest.raises(RuntimeError, match="You must first set the image array"):
-        im.get_shape()
+    # Initial behavior is an empty 2D array
+    assert im.get_shape() == (1, 0)
     z = np.arange(12, dtype=float).reshape((4, 3))
     im.set_data(z)
     assert im.get_shape() == (4, 3)
