支持1d级别插针模型

2026-08-25 09:23:30 +08:00
parent baddd7763d
commit c0df80154b
12 changed files with 299 additions and 65 deletions
--- a/tests/test_latest_predict.py
+++ b/tests/test_latest_predict.py
@@ -6,7 +6,7 @@ from unittest.mock import patch

 import pandas as pd

-from cbpred.latest_predict import OUTPUT_COLUMNS, build_predictions
+from cbpred.latest_predict import OUTPUT_COLUMNS, build_predictions, output_columns_for_timeframe


 class LatestPredictTests(unittest.TestCase):
@@ -27,6 +27,9 @@ 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_batch_skips_symbols_without_training_profiles(self):
        with tempfile.TemporaryDirectory() as directory:
            root = Path(directory)