Matplotlib backcompat in tests

mwaskom · Dec 6, 2023 · 872c7fc · 872c7fc
1 parent 785242b
commit 872c7fc
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_relational.py b/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)