From 9531fbc800be95bed430ddfbb512d45ed5f6e127 Mon Sep 17 00:00:00 2001
From: Shankhadeep Mandal <118059484+aenlemmea@users.noreply.github.com>
Date: Tue, 24 Oct 2023 00:18:03 +0530
Subject: [PATCH] Added tests for truncate_titles() (#232)

* Added tests for truncate_titles()

* Fixed formatting issues
---
 tests/test_viz_manager.py | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/tests/test_viz_manager.py b/tests/test_viz_manager.py
index 17529bbe..fa4a875a 100644
--- a/tests/test_viz_manager.py
+++ b/tests/test_viz_manager.py
@@ -1,7 +1,7 @@
 import pytest
 from PIL import Image
 
-from cleanvision.utils.viz_manager import VizManager
+from cleanvision.utils.viz_manager import VizManager, truncate_titles
 
 
 class TestVizManager:
@@ -30,3 +30,25 @@ def test_individual_images(self, images, title_info):
     )
     def test_image_sets(self, image_sets, title_info_sets):
         VizManager.image_sets(image_sets, title_info_sets, 4, (2, 2))
+
+
+def test_truncate_titles():
+    assert truncate_titles(
+        2,
+        [
+            "/home/usr/proj/dev/product/dataset/images/image_0001.img",
+            "/home/usr/proj/dev/product/dataset/images/image_0002.img",
+        ],
+    ) == ["...es/image_0001.img", "...es/image_0002.img"]
+
+    assert truncate_titles(2, ["image.jpeg", "image2.jpeg"]) == [
+        "image.jpeg",
+        "image2.jpeg",
+    ]
+    assert truncate_titles(
+        2,
+        [
+            "/pictures/mount/image_0001.img",
+            "/home/usr/proj/dev/product/dataset/images/image_0002.img",
+        ],
+    ) == ["/pictures/mount/i...", "/home/usr/proj/de..."]