[Wheel] Stage _structured_ranges.py into the combined wheel

python/tests/packaging/test_installed_wheel.py

@@ -44,6 +44,7 @@ def test_wheel_imports_outside_the_repository(
     with zipfile.ZipFile(wheel) as archive:
         wheel_files = set(archive.namelist())
     assert "mooncake/store/__init__.py" in wheel_files
+    assert "mooncake/_structured_ranges.py" in wheel_files
     assert {"mooncake/py.typed", "mooncake/engine.pyi"} <= wheel_files
     assert not any(name.endswith("mooncake_store_rs.pth") for name in wheel_files)
     assert not any(name.startswith("mooncake_store_rs/") for name in wheel_files)
@@ -198,8 +199,11 @@ def test_wheel_imports_outside_the_repository(
 assert util.find_spec("numpy") is None
 assert util.find_spec("PIL") is None
 assert util.find_spec("mooncake.structured_object_store") is not None
+assert util.find_spec("mooncake._structured_ranges") is not None
 assert "mooncake.structured_object_store" not in sys.modules
+assert "mooncake._structured_ranges" not in sys.modules
 assert "mooncake/structured_object_store.py" in installed_files
+assert "mooncake/_structured_ranges.py" in installed_files
 assert any(
     path.startswith("mooncake/_fast_copy.") and path.endswith(".so")
     for path in installed_files

python/tests/packaging/test_project.py

@@ -98,7 +98,11 @@ def test_structured_object_storage_has_one_authoritative_source() -> None:
     legacy_package_root = REPOSITORY_ROOT / "mooncake-wheel" / "mooncake"
     legacy_test_root = REPOSITORY_ROOT / "mooncake-wheel" / "tests"
 
-    for module in ("structured_object_store.py", "_fast_copy.c"):
+    for module in (
+        "structured_object_store.py",
+        "_structured_ranges.py",
+        "_fast_copy.c",
+    ):
         assert (package_root / module).is_file()
         assert not (legacy_package_root / module).exists()
 
@@ -115,7 +119,11 @@ def test_structured_object_storage_has_one_authoritative_source() -> None:
         .split(")", 1)[0]
         .split()
     )
-    assert {"structured_object_store.py", "_fast_copy.c"} <= set(migrated_modules)
+    assert {
+        "structured_object_store.py",
+        "_structured_ranges.py",
+        "_fast_copy.c",
+    } <= set(migrated_modules)
 
 
 def test_http_metadata_service_has_one_authoritative_source(

scripts/build_wheel.sh

@@ -205,6 +205,7 @@ MIGRATED_PYTHON_MODULES=(
     py.typed
     engine.pyi
     structured_object_store.py
+    _structured_ranges.py
     _fast_copy.c
     http_metadata_server.py
     _launcher.py