update

tests/test_latest_predict.py

@@ -6,7 +6,13 @@ from unittest.mock import patch
 
 import pandas as pd
 
-from cbpred.latest_predict import OUTPUT_COLUMNS, build_predictions, output_columns_for_timeframe
+from cbpred.latest_predict import (
+    OUTPUT_COLUMNS,
+    REPO_ROOT,
+    build_predictions,
+    default_output_dir,
+    output_columns_for_timeframe,
+)
 
 
 class LatestPredictTests(unittest.TestCase):
@@ -30,6 +36,10 @@ class LatestPredictTests(unittest.TestCase):
     def test_output_columns_follow_timeframe(self):
         self.assertEqual(output_columns_for_timeframe("1d")[4], "quote_volume_usdt_1d")
 
+    def test_default_output_dir_follows_timeframe(self):
+        self.assertEqual(default_output_dir("8h"), REPO_ROOT / "predictions/latest_v1_8h")
+        self.assertEqual(default_output_dir("1d"), REPO_ROOT / "predictions/latest_v1_1d")
+
     def test_batch_skips_symbols_without_training_profiles(self):
         with tempfile.TemporaryDirectory() as directory:
             root = Path(directory)