Skip to content

Commit

Permalink
Update plot_legend test
Browse files Browse the repository at this point in the history
  • Loading branch information
breimanntools committed Apr 9, 2024
1 parent b523b64 commit e169de7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/unit/plotting_tests/test_plot_legend.py
Original file line number Diff line number Diff line change
Expand Up @@ -166,7 +166,7 @@ def test_plot_set_legend_n_cols(self, n_cols):
@given(st.lists(st.text(), min_size=2, max_size=5))
def test_plot_set_legend_custom_labels(self, labels):
fig, ax = plt.subplots()
labels = list(set(labels))
labels = list(set([x for x in labels if len(x) > 1]))
ax = aa.plot_legend(ax=ax, dict_color={label: "red" for label in labels})
legend_labels = [text.get_text() for text in ax.get_legend().get_texts()]
assert set(legend_labels) == set(labels)
Expand Down

0 comments on commit e169de7

Please sign in to comment.