def test_dry_run(manager: storage.StorageManager, to_delete: List[str]) -> None:
delete_checkpoints(manager, to_delete, dry_run=True)
assert len(os.listdir(manager._base_path)) == len(to_delete)
def test_validate_failure(config: Dict[str, Any], to_delete: List[Dict[str, Any]]) -> None:
host_path = config["checkpoint_storage"].pop("host_path")
with pytest.raises(TypeError):
delete_checkpoints(config, to_delete, validate=True, dry_run=False)
assert len(os.listdir(host_path)) == len(to_delete)
def test_delete_checkpoints(manager: storage.StorageManager,
to_delete: List[Dict[str, Any]]) -> None:
delete_checkpoints(manager, to_delete, dry_run=False)
assert len(os.listdir(manager._base_path)) == 0
def test_validate_success(config: Dict[str, Any], to_delete: List[Dict[str, Any]]) -> None:
delete_checkpoints(config, to_delete, validate=True, dry_run=False)
host_path = config["checkpoint_storage"]["host_path"]
assert len(os.listdir(host_path)) == 0