Fixed error in drawing last label

LucaCappelletti94 · Feb 29, 2020 · 7be4466 · 7be4466
1 parent b507263
commit 7be4466
Show file tree

Hide file tree

Showing 18 changed files with 42 additions and 0 deletions.
diff --git a/barplots/utils/text_positions.py b/barplots/utils/text_positions.py
@@ -33,6 +33,8 @@ def text_positions(df: pd.DataFrame, bar_width: float, index_level: int) -> Gene
                     yield text_position/2 + bar_width/2, old_index[j]
         if j == index_level and j == len(jumps)-1 and is_last(df, i):
             yield text_position/2 + bar_width*1.5, index[j]
+        if index_level==0 and is_last(df, i):
+            yield text_position/2, index[index_level]
         old_index = index
         for j, value in enumerate(jumps):
             if value:

diff --git a/examples/horizontal_legend_minor_rotation_val_auroc.png b/examples/horizontal_legend_minor_rotation_val_auroc.png
diff --git a/examples/horizontal_legend_subplots_minor_rotation_val_auroc.png b/examples/horizontal_legend_subplots_minor_rotation_val_auroc.png
diff --git a/examples/horizontal_major_rotation_val_auroc.png b/examples/horizontal_major_rotation_val_auroc.png
diff --git a/examples/horizontal_subplots_major_rotation_unique_major_labels_val_auroc.png b/examples/horizontal_subplots_major_rotation_unique_major_labels_val_auroc.png
diff --git a/...l_subplots_major_rotation_unique_minor_labels_unique_major_labels_val_auroc.png b/...l_subplots_major_rotation_unique_minor_labels_unique_major_labels_val_auroc.png
diff --git a/examples/horizontal_subplots_major_rotation_unique_minor_labels_val_auroc.png b/examples/horizontal_subplots_major_rotation_unique_minor_labels_val_auroc.png
diff --git a/examples/horizontal_subplots_major_rotation_val_auroc.png b/examples/horizontal_subplots_major_rotation_val_auroc.png
diff --git a/examples/vertical_legend_subplots_val_auroc.png b/examples/vertical_legend_subplots_val_auroc.png
diff --git a/examples/vertical_legend_val_auroc.png b/examples/vertical_legend_val_auroc.png
diff --git a/examples/vertical_minor_rotation_val_auroc.png b/examples/vertical_minor_rotation_val_auroc.png
diff --git a/examples/vertical_subplots_minor_rotation_unique_major_labels_val_auroc.png b/examples/vertical_subplots_minor_rotation_unique_major_labels_val_auroc.png
diff --git a/...l_subplots_minor_rotation_unique_minor_labels_unique_major_labels_val_auroc.png b/...l_subplots_minor_rotation_unique_minor_labels_unique_major_labels_val_auroc.png
diff --git a/examples/vertical_subplots_minor_rotation_unique_minor_labels_val_auroc.png b/examples/vertical_subplots_minor_rotation_unique_minor_labels_val_auroc.png
diff --git a/examples/vertical_subplots_minor_rotation_val_auroc.png b/examples/vertical_subplots_minor_rotation_val_auroc.png
diff --git a/test_barplots/val_auroc.png b/test_barplots/val_auroc.png
diff --git a/tests/test_simple_barplots.py b/tests/test_simple_barplots.py
@@ -0,0 +1,27 @@
+import pandas as pd
+from barplots import barplots
+
+
+def test_simple_barplots():
+    root = "test_barplots"
+    df = pd.read_csv("tests/test_case.csv")
+
+    custom_defaults = {
+        "P": "promoters",
+        "E": "enhancers",
+        "A": "active ",
+        "I": "inactive ",
+        "+": " and ",
+        "": "anything",
+        "Validation": "val"
+    }
+
+    barplots(
+        df,
+        ["cell_line", "task", "model"],
+        path="{root}/{{feature}}.png".format(root=root),
+        orientation="horizontal",
+        verbose=False,
+        custom_defaults=custom_defaults,
+        subplots=True
+    )
diff --git a/tests/test_single.py b/tests/test_single.py
@@ -0,0 +1,13 @@
+import pandas as pd
+from barplots import barplots
+
+
+def test_single_index():
+    root = "test_barplots"
+    df = pd.read_csv("tests/test_case.csv")
+    barplots(
+        df,
+        ["cell_line"],
+        path="{root}/{{feature}}.png".format(root=root),
+        verbose=False
+    )