fix: resolve the case if the adjacent entity paths are duplicate (#165)

Signed-off-by: ktro2828 <kotaro.uetake@tier4.jp>

Author: ktro2828

t4_devkit/viewer/viewer.py

@@ -30,11 +30,10 @@
 __all__ = ["RerunViewer", "format_entity"]
 
 
-def format_entity(root: str, *entities: Sequence[str]) -> str:
+def format_entity(*entities: Sequence[str]) -> str:
     """Format entity path.
 
     Args:
-        root (str): Root entity path.
         *entities: Entity path(s).
 
     Returns:
@@ -48,15 +47,16 @@ def format_entity(root: str, *entities: Sequence[str]) -> str:
         >>> format_entity("map", "map/base_link", "camera")
         "map/base_link/camera"
     """
-    if len(entities) == 0:
-        return root
-
-    flattened = [s for t in entities for s in t.split("/")]
-
-    if osp.basename(root) == flattened[0]:
-        return osp.join(root, "/".join(flattened[1:])) if len(flattened) > 1 else root
-    else:
-        return osp.join(root, "/".join(entities))
+    if not entities:
+        return ""
+
+    flattened = []
+    for entity in entities:
+        for part in entity.split("/"):
+            if part and flattened and flattened[-1] == part:
+                continue
+            flattened.append(part)
+    return "/".join(flattened)
 
 
 class RerunViewer:

tests/viewer/test_viewer.py

@@ -14,6 +14,7 @@ def test_format_entity() -> None:
     assert "map/base_link" == format_entity("map", "base_link")
     assert "map/base_link" == format_entity("map", "map/base_link")
     assert "map/base_link/camera" == format_entity("map", "map/base_link", "camera")
+    assert "map/base_link/camera" == format_entity("map", "map/base_link", "base_link/camera")
 
 
 def test_render_box3ds(dummy_viewer, dummy_box3ds) -> None: