feat(selection): 集成 B1 FastDTW 图形评分
This commit is contained in:
@@ -26,7 +26,8 @@ def test_postgres_migration_creates_market_data_contract(
|
||||
engine: Engine = create_engine(sqlalchemy_url)
|
||||
command.upgrade(config, "head")
|
||||
try:
|
||||
tables = set(inspect(engine).get_table_names())
|
||||
inspector = inspect(engine)
|
||||
tables = set(inspector.get_table_names())
|
||||
assert {
|
||||
"market_stock",
|
||||
"market_daily_bar",
|
||||
@@ -39,6 +40,25 @@ def test_postgres_migration_creates_market_data_contract(
|
||||
"selection_run_item",
|
||||
"selection_signal",
|
||||
} <= tables
|
||||
item_columns = {column["name"] for column in inspector.get_columns("selection_run_item")}
|
||||
assert {
|
||||
"score_status",
|
||||
"score_value",
|
||||
"score_threshold",
|
||||
"score_version",
|
||||
"match_case_id",
|
||||
"match_case_name",
|
||||
"match_case_breakout_date",
|
||||
"match_breakdown",
|
||||
"score_reason",
|
||||
} <= item_columns
|
||||
constraint_names = {
|
||||
constraint["name"]
|
||||
for constraint in inspector.get_check_constraints("selection_run_item")
|
||||
}
|
||||
assert "ck_selection_run_item_breakdown_range" in constraint_names
|
||||
index_names = {index["name"] for index in inspector.get_indexes("selection_run_item")}
|
||||
assert "ix_selection_run_item_score" in index_names
|
||||
finally:
|
||||
engine.dispose()
|
||||
get_settings.cache_clear()
|
||||
|
||||
Reference in New Issue
Block a user