simonandClaude Opus 4.7
271a9343a5
7 Sprints 全部完成: Sprint 0: 基础设施 (DataManager + MariaDB) Sprint 1: 因子引擎 (34因子/12分类) Sprint 2: VectorBT 回测 (5策略+截面) Sprint 3: Optuna 优化 (+Walk-Forward) Sprint 4: ML 模型 (LightGBM+CatBoost) Sprint 5: Qwen 情绪因子 (三源新闻+日期对齐) Sprint 6: Agent 系统 (4Agent+日报.md/.html) 生产加固 (15项): Tushare双源fallback, SSH自动恢复, pool_pre_ping, save_daily先删后插, load_dotenv绝对路径, 日报5d/20d修复, RiskAgent改上证指数, 昨日对比+数据截止, mac_report utf8mb4, CLAUDE-*.md 9条已知Bug, demo全参数化, djapi数据源归一化, indexDatas API修正 Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
88 lines
2.8 KiB
Python
88 lines
2.8 KiB
Python
from django.test import TestCase
|
|
from .stock.stock_utils import tscodeCheck, date_format_correction, get_released_report_dates
|
|
|
|
|
|
class TscodeCheckTest(TestCase):
|
|
"""股票代码格式校验测试"""
|
|
|
|
def test_sz_6digit(self):
|
|
self.assertEqual(tscodeCheck('000001'), '000001.SZ')
|
|
|
|
def test_sz_6digit_300(self):
|
|
self.assertEqual(tscodeCheck('300750'), '300750.SZ')
|
|
|
|
def test_sh_6digit_60(self):
|
|
self.assertEqual(tscodeCheck('600000'), '600000.SH')
|
|
|
|
def test_sh_6digit_68(self):
|
|
self.assertEqual(tscodeCheck('688001'), '688001.SH')
|
|
|
|
def test_bj_6digit(self):
|
|
self.assertEqual(tscodeCheck('830799'), '830799.BJ')
|
|
|
|
def test_9digit_pass_through(self):
|
|
self.assertEqual(tscodeCheck('000001.SZ'), '000001.SZ')
|
|
|
|
def test_lowercase_to_uppercase(self):
|
|
self.assertEqual(tscodeCheck('000001.sz'), '000001.SZ')
|
|
|
|
def test_invalid_length_short(self):
|
|
with self.assertRaises(ValueError):
|
|
tscodeCheck('12345')
|
|
|
|
def test_invalid_length_long(self):
|
|
with self.assertRaises(ValueError):
|
|
tscodeCheck('1234567890')
|
|
|
|
def test_invalid_suffix(self):
|
|
with self.assertRaises(ValueError):
|
|
tscodeCheck('000001.XX')
|
|
|
|
def test_not_string(self):
|
|
with self.assertRaises(ValueError):
|
|
tscodeCheck(123456)
|
|
|
|
def test_unknown_prefix(self):
|
|
with self.assertRaises(ValueError):
|
|
tscodeCheck('500001')
|
|
|
|
|
|
class DateFormatCorrectionTest(TestCase):
|
|
"""日期格式校正测试"""
|
|
|
|
def test_yyyymmdd_passthrough(self):
|
|
self.assertEqual(date_format_correction('20240115'), '20240115')
|
|
|
|
def test_yyyy_mm_dd_conversion(self):
|
|
self.assertEqual(date_format_correction('2024-01-15'), '20240115')
|
|
|
|
def test_invalid_format(self):
|
|
self.assertIsNone(date_format_correction('15/01/2024'))
|
|
|
|
def test_empty_string(self):
|
|
self.assertIsNone(date_format_correction(''))
|
|
|
|
|
|
class GetReleasedReportDatesTest(TestCase):
|
|
"""财报发布日期测试"""
|
|
|
|
def test_returns_list(self):
|
|
result = get_released_report_dates('20230101', '20231231')
|
|
self.assertIsInstance(result, list)
|
|
|
|
def test_all_dates_yyyymmdd_format(self):
|
|
result = get_released_report_dates('20230101', '20231231')
|
|
for d in result:
|
|
self.assertEqual(len(d), 8)
|
|
self.assertTrue(d.isdigit())
|
|
|
|
def test_start_after_end_returns_empty(self):
|
|
result = get_released_report_dates('20251231', '20230101')
|
|
self.assertEqual(result, [])
|
|
|
|
def test_single_year_quarters(self):
|
|
"""单年内应该返回最多4个季末日期"""
|
|
result = get_released_report_dates('20200101', '20201231')
|
|
for d in result:
|
|
self.assertTrue(d.endswith(('0331', '0630', '0930', '1231')))
|