Skip to content

Commit eb76c60

Browse files
Copilotxadupre
andauthored
Add extra validation tests for waterfall helper
Agent-Logs-Url: https://github.com/sdpython/teachpyx/sessions/e6e8e7c7-c0a1-47b3-9763-5ebb04affcb9 Co-authored-by: xadupre <22452781+xadupre@users.noreply.github.com>
1 parent 0d5d398 commit eb76c60

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

_unittests/ut_tools/test_pandas.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,17 @@ def test_plot_waterfall_missing_column(self):
3232
df = pandas.DataFrame({"name": ["A"], "delta": [1]})
3333
self.assertRaise(lambda: plot_waterfall(df, "missing", "name"), ValueError)
3434

35+
def test_plot_waterfall_missing_label_column(self):
36+
df = pandas.DataFrame({"name": ["A"], "delta": [1]})
37+
self.assertRaise(lambda: plot_waterfall(df, "delta", "missing"), ValueError)
38+
39+
def test_plot_waterfall_bad_colors(self):
40+
df = pandas.DataFrame({"name": ["A"], "delta": [1]})
41+
self.assertRaise(
42+
lambda: plot_waterfall(df, "delta", "name", colors=("r",)),
43+
ValueError,
44+
)
45+
3546

3647
if __name__ == "__main__":
3748
unittest.main()

0 commit comments

Comments
 (0)