Add wrong y_labels error.

pyranges · Jun 14, 2024 · b0bdb56 · b0bdb56
1 parent 8b7ac60
commit b0bdb56
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/pyranges_plot/plot_main.py b/src/pyranges_plot/plot_main.py
@@ -165,6 +165,13 @@ def plot(
     if not isinstance(data, list):
         data = [data]
 
+    # Ensure correct y_labels
+    if y_labels:
+        if len(y_labels) != len(data):
+            raise Exception(
+                f"The number of provided y_labels {y_labels} does not match the number of PyRanges objects ({len(data)})."
+            )
+
     # Deal with export
     if to_file:
         # given str file name