Fix file glob not correctly reading sdfs with prefixed files

Author: JoelLucaAdams

src/sdf_xarray/__init__.py

@@ -103,7 +103,7 @@ def _resolve_glob(path_glob: PathLike | Iterable[PathLike]):
 
     try:
         p = Path(path_glob)
-        paths = list(p.parent.glob(p.name)) if p.name == "*.sdf" else list(p)
+        paths = list(p.parent.glob(p.name)) if p.name.endswith("*.sdf") else list(p)
     except TypeError:
         paths = list({Path(p) for p in path_glob})
 

tests/test_dataset.py

@@ -212,6 +212,18 @@ def test_resolve_glob_from_string_pattern():
     assert result == expected
 
 
+def test_resolve_glob_from_multiple_names_string_pattern(tmp_path):
+    mock_test_files_dir = tmp_path
+    (mock_test_files_dir / "normal_0000.sdf").touch()
+    (mock_test_files_dir / "normal_0001.sdf").touch()
+    (mock_test_files_dir / "other_0000.sdf").touch()
+
+    pattern = str(mock_test_files_dir / "normal_*.sdf")
+    result = _resolve_glob(pattern)
+    expected = sorted(mock_test_files_dir.glob("normal_*.sdf"))
+    assert result == expected
+
+
 def test_resolve_glob_from_path_glob():
     pattern = TEST_FILES_DIR.glob("*.sdf")
     result = _resolve_glob(pattern)