Test get_upath_for_protocol is only called once

astronomy-commons · Mar 4, 2025 · 246b1be · 246b1be
1 parent 458b040
commit 246b1be
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/hats/io/file_io/__init__.py b/src/hats/io/file_io/__init__.py
@@ -22,5 +22,6 @@
     find_files_matching_path,
     get_directory_contents,
     get_upath,
+    get_upath_for_protocol,
     is_regular_file,
 )
diff --git a/tests/hats/catalog/loaders/test_read_hats.py b/tests/hats/catalog/loaders/test_read_hats.py
@@ -1,3 +1,4 @@
+from hats.io.file_io import get_upath_for_protocol
 from hats.loaders import read_hats
 
 
@@ -17,3 +18,10 @@ def test_read_hats_branches(
     read_hats(margin_catalog_path)
     read_hats(small_sky_source_dir)
     read_hats(test_data_dir / "square_map")
+
+
+def test_read_hats_initializes_upath_once(small_sky_dir, mocker):
+    mock_method = "hats.io.file_io.file_pointer.get_upath_for_protocol"
+    mocked_upath_call = mocker.patch(mock_method, side_effect=get_upath_for_protocol)
+    read_hats(small_sky_dir)
+    mocked_upath_call.assert_called_once_with(small_sky_dir)