Matplotlib backcompat in tests

Author: mwaskom

tests/test_relational.py

@@ -582,8 +582,8 @@ def test_relplot_weighted_estimator(self, long_df):
 
         g = relplot(data=long_df, x="a", y="y", weights="x", kind="line")
         ydata = g.ax.lines[0].get_ydata()
-        for i, label in enumerate(g.ax.get_xticklabels()):
-            pos_df = long_df[long_df["a"] == label.get_text()]
+        for i, level in enumerate(categorical_order(long_df["a"])):
+            pos_df = long_df[long_df["a"] == level]
             expected = np.average(pos_df["y"], weights=pos_df["x"])
             assert ydata[i] == pytest.approx(expected)
 
@@ -1072,8 +1072,8 @@ def test_weights(self, long_df):
 
         ax = lineplot(long_df, x="a", y="y", weights="x")
         vals = ax.lines[0].get_ydata()
-        for i, label in enumerate(ax.get_xticklabels()):
-            pos_df = long_df.loc[long_df["a"] == label.get_text()]
+        for i, level in enumerate(categorical_order(long_df["a"])):
+            pos_df = long_df[long_df["a"] == level]
             expected = np.average(pos_df["y"], weights=pos_df["x"])
             assert vals[i] == pytest.approx(expected)