Raise explicit ValueError for empty densities argument

tpvasconcelos · Apr 19, 2024 · cf69c73 · cf69c73
1 parent 302423c
commit cf69c73
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/ridgeplot/_figure_factory.py b/src/ridgeplot/_figure_factory.py
@@ -127,6 +127,8 @@ def get_xy_extrema(densities: Densities) -> tuple[Numeric, Numeric, Numeric, Num
     ... )
     (-2, 4, 0, 4)
     """
+    if len(densities) == 0:
+        raise ValueError("The densities array should not be empty.")
     x_flat: list[Numeric] = []
     y_flat: list[Numeric] = []
     for row in densities:

diff --git a/tests/unit/test_figure_factory.py b/tests/unit/test_figure_factory.py
@@ -24,7 +24,7 @@ class TestGetXYExtrema:
 
     def test_raise_for_empty_sequence(self) -> None:
         # Fails for empty sequence
-        with pytest.raises(ValueError, match="arg is an empty sequence"):
+        with pytest.raises(ValueError, match="The densities array should not be empty"):
             get_xy_extrema(densities=[])
 
     def test_raise_for_non_2d_array(self) -> None: