Add test

TLCFEM · Mar 16, 2024 · 5653783 · 5653783
1 parent d43ada4
commit 5653783
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/msglc/reader.py b/src/msglc/reader.py
@@ -443,6 +443,8 @@ def read(self, path: str | list | slice | None = None):
 
         target = self._obj
         for key in path_stack:
+            if "" == key:
+                continue
             if isinstance(key, str) and isinstance(target, (list, LazyList)):
                 key = to_index(key, len(target))
             target = target[key]
@@ -462,7 +464,7 @@ def visit(self, path: str = ""):
         """
         target = self._obj
         for key in path.split("/"):
-            if not key:
+            if "" == key:
                 continue
             if isinstance(target, (list, LazyList)):
                 key = to_index(key, len(target))

diff --git a/tests/test_msglc.py b/tests/test_msglc.py
@@ -169,7 +169,7 @@ def test_combine_archives(tmpdir, json_after, target):
             target.seek(0)
 
         with LazyReader(target) as reader:
-            assert reader.read("first_outer/second_inner/glossary/title") == "example glossary"
+            assert reader.read("first_outer//second_inner/glossary/title") == "example glossary"
             assert reader.read("second_outer/first_inner/2") == 2
             assert reader.read("second_outer/first_inner/-1") == 29
             assert reader.read("second_outer/first_inner/0:2") == [0, 1]
@@ -178,7 +178,7 @@ def test_combine_archives(tmpdir, json_after, target):
             assert reader.read("second_outer/first_inner/24:2:30") == [24, 26, 28]
             assert reader.read("second_outer/first_inner/:2:5") == [0, 2, 4]
             assert reader.read("second_outer/first_inner/24:2:") == [24, 26, 28]
-            assert reader.visit("first_outer/second_inner/glossary/title") == "example glossary"
+            assert reader.visit("first_outer//second_inner/glossary/title") == "example glossary"
             assert reader.visit("second_outer/first_inner/2") == 2
             assert reader.visit("second_outer/first_inner/-1") == 29
             assert reader.visit("second_outer/first_inner/0:2") == [0, 1]