Pass now to _prune_old_files for deterministic boundary testing

Mirrors the same pattern used by _build_table. Tests now use a fixed
NOW constant and relative offsets, enabling precise boundary checks
(exactly keep_days old is kept, keep_days+1 is deleted).

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

logging_utils.py

@@ -96,10 +96,10 @@ def _upload_parquet(api, repo_id, table, path_in_repo):
                 pass
 
 
-def _prune_old_files(api, repo_id, keep_days):
+def _prune_old_files(api, repo_id, keep_days, now):
     if keep_days <= 0:
         return
-    cutoff = (datetime.now(timezone.utc) - timedelta(days=keep_days)).strftime("%Y-%m-%d")
+    cutoff = (now - timedelta(days=keep_days)).strftime("%Y-%m-%d")
     try:
         to_delete = [
             f.path
@@ -133,7 +133,7 @@ def log_inference(pil_inputs, output_pil, prompt, seed, steps, guidance_scale,
         api = HfApi(token=HF_TOKEN)
         api.create_repo(repo_id=DATASET_REPO, repo_type="dataset", private=True, exist_ok=True)
         _upload_parquet(api, DATASET_REPO, table, path_in_repo)
-        _prune_old_files(api, DATASET_REPO, MAX_LOG_DAYS)
+        _prune_old_files(api, DATASET_REPO, MAX_LOG_DAYS, now)
     except Exception as log_err:
         import traceback as _tb
         print(f"[log] WARNING: {log_err}\n{_tb.format_exc()}")

tests/test_logging_utils.py

@@ -1,68 +1,78 @@
 from datetime import datetime, timezone, timedelta
 from types import SimpleNamespace
-from unittest.mock import MagicMock, call
+from unittest.mock import MagicMock
 
 from logging_utils import _prune_old_files
 
+NOW = datetime(2026, 5, 17, 12, 0, 0, tzinfo=timezone.utc)
+KEEP_DAYS = 7
+
 
 def _make_api(paths):
-    """Return a mock HfApi whose list_repo_tree yields RepoFile-like objects for `paths`."""
     api = MagicMock()
     api.list_repo_tree.return_value = [SimpleNamespace(path=p) for p in paths]
     return api
 
 
-def _old(name):
-    return f"data/2020-01-01-000000-{name}.parquet"
+def _file(days_ago):
+    date = (NOW - timedelta(days=days_ago)).strftime("%Y-%m-%d")
+    return f"data/{date}-000000-abcd1234.parquet"
 
 
-def _recent(name):
-    today = datetime.now(timezone.utc).strftime("%Y-%m-%d")
-    return f"data/{today}-000000-{name}.parquet"
+def _deleted_paths(api):
+    return {c.kwargs["path_in_repo"] for c in api.delete_file.call_args_list}
 
 
 def test_old_files_are_deleted():
-    api = _make_api([_old("aaa"), _old("bbb")])
-    _prune_old_files(api, "org/repo", keep_days=7)
-    assert api.delete_file.call_count == 2
-    deleted = {c.kwargs["path_in_repo"] for c in api.delete_file.call_args_list}
-    assert deleted == {_old("aaa"), _old("bbb")}
+    api = _make_api([_file(8), _file(10)])
+    _prune_old_files(api, "org/repo", KEEP_DAYS, NOW)
+    assert _deleted_paths(api) == {_file(8), _file(10)}
 
 
 def test_recent_files_are_kept():
-    api = _make_api([_recent("aaa"), _recent("bbb")])
-    _prune_old_files(api, "org/repo", keep_days=7)
+    api = _make_api([_file(6), _file(1), _file(0)])
+    _prune_old_files(api, "org/repo", KEEP_DAYS, NOW)
+    api.delete_file.assert_not_called()
+
+
+def test_boundary_exactly_keep_days_old_is_kept():
+    api = _make_api([_file(KEEP_DAYS)])
+    _prune_old_files(api, "org/repo", KEEP_DAYS, NOW)
     api.delete_file.assert_not_called()
 
 
+def test_one_day_past_boundary_is_deleted():
+    api = _make_api([_file(KEEP_DAYS + 1)])
+    _prune_old_files(api, "org/repo", KEEP_DAYS, NOW)
+    assert _deleted_paths(api) == {_file(KEEP_DAYS + 1)}
+
+
 def test_only_old_files_deleted_in_mixed_set():
-    api = _make_api([_old("aaa"), _recent("bbb"), _old("ccc")])
-    _prune_old_files(api, "org/repo", keep_days=7)
-    deleted = {c.kwargs["path_in_repo"] for c in api.delete_file.call_args_list}
-    assert deleted == {_old("aaa"), _old("ccc")}
+    api = _make_api([_file(8), _file(6), _file(10), _file(0)])
+    _prune_old_files(api, "org/repo", KEEP_DAYS, NOW)
+    assert _deleted_paths(api) == {_file(8), _file(10)}
 
 
 def test_empty_directory_does_nothing():
     api = _make_api([])
-    _prune_old_files(api, "org/repo", keep_days=7)
+    _prune_old_files(api, "org/repo", KEEP_DAYS, NOW)
     api.delete_file.assert_not_called()
 
 
 def test_keep_days_zero_skips_pruning():
-    api = _make_api([_old("aaa")])
-    _prune_old_files(api, "org/repo", keep_days=0)
+    api = _make_api([_file(8)])
+    _prune_old_files(api, "org/repo", keep_days=0, now=NOW)
     api.list_repo_tree.assert_not_called()
     api.delete_file.assert_not_called()
 
 
 def test_non_parquet_files_are_ignored():
-    api = _make_api(["data/2020-01-01-README.md", _old("aaa")])
-    _prune_old_files(api, "org/repo", keep_days=7)
-    deleted = {c.kwargs["path_in_repo"] for c in api.delete_file.call_args_list}
-    assert deleted == {_old("aaa")}
+    api = _make_api(["data/2020-01-01-README.md", _file(8)])
+    _prune_old_files(api, "org/repo", KEEP_DAYS, NOW)
+    assert _deleted_paths(api) == {_file(8)}
 
 
 def test_api_error_does_not_raise():
     api = MagicMock()
     api.list_repo_tree.side_effect = RuntimeError("network error")
-    _prune_old_files(api, "org/repo", keep_days=7)  # must not raise
+    _prune_old_files(api, "org/repo", KEEP_DAYS, NOW)  # must not raise