Fix testing to properly support rc releases

of matplotlib. Remove depreciated apis from
tests.

Author: isaacrobinson2000

matplotview/tests/test_view_obj.py

@@ -10,16 +10,16 @@ def test_obj_comparison():
     from matplotlib.axes import Subplot, Axes
     import matplotlib
 
-    mpl_version = tuple(int(v) for v in matplotlib.__version__.split("."))
+    mpl_version = tuple(int(v) for v in matplotlib.__version__.split(".")[:2])
 
     view_class1 = view_wrapper(Subplot)
     view_class2 = view_wrapper(Subplot)
     view_class3 = view_wrapper(Axes)
 
     assert view_class1 is view_class2
     assert view_class1 == view_class2
-    if(mpl_version >= (3, 7, 0)):
-        # As of 3.7.0, the subplot class no long exists, and is an alias to the Axes class...
+    if(mpl_version >= (3, 7)):
+        # As of 3.7.0, the subplot class no longer exists, and is an alias to the Axes class...
         assert view_class2 == view_class3
     else:
         assert view_class2 != view_class3

matplotview/tests/utils.py

@@ -4,8 +4,8 @@
 
 def figure_to_image(figure):
     figure.canvas.draw()
-    img = np.frombuffer(figure.canvas.tostring_rgb(), dtype=np.uint8)
-    return img.reshape(figure.canvas.get_width_height()[::-1] + (3,))
+    img = np.frombuffer(figure.canvas.buffer_rgba(), dtype=np.uint8)
+    return img.reshape(figure.canvas.get_width_height()[::-1] + (4,))[..., :3]
 
 
 def matches_post_pickle(figure):