Add checkpoint/_src/testing/multiprocess_test.py, checkpoint/_src/che…

…ckpointers/checkpointer_test_utils.py, checkpoint/checkpoint_manager_options_test.py.

PiperOrigin-RevId: 721519534

.github/workflows/build.yml

@@ -60,7 +60,7 @@ jobs:
     - name: Test with pytest
     # TODO(yaning): Move these to an exclude target within pytest.ini.
       run: |
-        python -m pytest --ignore=orbax/checkpoint/experimental/emergency/broadcast_multislice_test.py --ignore=orbax/checkpoint/experimental/emergency/checkpoint_manager_test.py --ignore=orbax/checkpoint/experimental/emergency/local_checkpoint_data_debugging_test.py --ignore=orbax/checkpoint/experimental/emergency/local_checkpoint_manager_test.py --ignore=orbax/checkpoint/experimental/emergency/multihost_test.py --ignore=orbax/checkpoint/experimental/emergency/replicator_checkpoint_manager_test.py
+        python -m pytest --ignore=orbax/checkpoint/experimental/emergency/broadcast_multislice_test.py --ignore=orbax/checkpoint/experimental/emergency/checkpoint_manager_test.py --ignore=orbax/checkpoint/experimental/emergency/local_checkpoint_data_debugging_test.py --ignore=orbax/checkpoint/experimental/emergency/local_checkpoint_manager_test.py --ignore=orbax/checkpoint/experimental/emergency/multihost_test.py --ignore=orbax/checkpoint/experimental/emergency/replicator_checkpoint_manager_test.py --ignore=orbax/checkpoint/_src/testing/multiprocess_test.py
     # The below step just reports the success or failure of tests as a "commit status".
     # This is needed for copybara integration.
     - name: Report success or failure as github status

checkpoint/orbax/checkpoint/BUILD

@@ -226,6 +226,12 @@ py_test(
     deps = [":msgpack_utils"],
 )
 
+py_test(
+    name = "checkpoint_manager_options_test",
+    srcs = ["checkpoint_manager_options_test.py"],
+    deps = [":checkpoint"],
+)
+
 py_test(
     name = "checkpoint_utils_test",
     srcs = ["checkpoint_utils_test.py"],

checkpoint/orbax/checkpoint/_src/checkpointers/BUILD

@@ -68,3 +68,31 @@ py_library(
         "//checkpoint/orbax/checkpoint/_src/path:atomicity_types",
     ],
 )
+
+py_library(
+    name = "checkpointer_test_utils",
+    srcs = ["checkpointer_test_utils.py"],
+    deps = [
+        ":async_checkpointer",
+        ":checkpointer",
+        "//checkpoint/orbax/checkpoint:args",
+        "//checkpoint/orbax/checkpoint:test_utils",
+        "//checkpoint/orbax/checkpoint:utils",
+        "//checkpoint/orbax/checkpoint/_src:asyncio_utils",
+        "//checkpoint/orbax/checkpoint/_src/handlers:async_checkpoint_handler",
+        "//checkpoint/orbax/checkpoint/_src/handlers:composite_checkpoint_handler",
+        "//checkpoint/orbax/checkpoint/_src/handlers:pytree_checkpoint_handler",
+        "//checkpoint/orbax/checkpoint/_src/logging:step_statistics",
+        "//checkpoint/orbax/checkpoint/_src/metadata:array_metadata",
+        "//checkpoint/orbax/checkpoint/_src/metadata:array_metadata_store",
+        "//checkpoint/orbax/checkpoint/_src/metadata:checkpoint",
+        "//checkpoint/orbax/checkpoint/_src/metadata:step_metadata_serialization",
+        "//checkpoint/orbax/checkpoint/_src/metadata:tree",
+        "//checkpoint/orbax/checkpoint/_src/multihost",
+        "//checkpoint/orbax/checkpoint/_src/path:atomicity",
+        "//checkpoint/orbax/checkpoint/_src/path:step",
+        "//checkpoint/orbax/checkpoint/_src/serialization",
+        "//checkpoint/orbax/checkpoint/_src/serialization:type_handlers",
+        "//checkpoint/orbax/checkpoint/_src/testing:test_tree_utils",
+    ],
+)