zsb/fidc-backtest-engine
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修正FiRisk强平触发来源

Browse Source
This commit is contained in:
boris
2026-06-20 23:07:59 +08:00
parent cdca7984ed
commit f8bc0679ee
11 changed files with 189 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/fidc-core/src/broker.rs
+2
View File
@@ -5458,6 +5458,7 @@ mod tests {
allow_sell,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}
}
@@ -5706,6 +5707,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
crates/fidc-core/src/data.rs
+10
View File
@@ -208,6 +208,8 @@ pub struct CandidateEligibility {
pub allow_sell: bool,
pub is_kcb: bool,
pub is_one_yuan: bool,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub risk_level_code: Option<String>,
}
impl CandidateEligibility {
@@ -3126,6 +3128,13 @@ fn read_candidates(path: &Path) -> Result<Vec<CandidateEligibility>, DataSetErro
allow_sell: row.parse_bool(6)?,
is_kcb: row.parse_optional_bool(7).unwrap_or(false),
is_one_yuan: row.parse_optional_bool(8).unwrap_or(false),
risk_level_code: row
.fields
.get(9)
.map(String::as_str)
.map(str::trim)
.filter(|value| !value.is_empty())
.map(ToOwned::to_owned),
});
}
Ok(snapshots)
@@ -3954,6 +3963,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
};
let data = DataSet::from_components(
vec![instrument("000001.SZ"), instrument("000002.SZ")],
crates/fidc-core/src/platform_expr_strategy.rs
+100 -2
View File
@@ -5956,7 +5956,15 @@ impl PlatformExprStrategy {
let Ok(candidate) = ctx.data.require_candidate(date, symbol) else {
return Ok(false);
};
if candidate.allow_buy || !candidate.allow_sell || candidate.is_paused {
let Some(_risk_level_code) = candidate
.risk_level_code
.as_deref()
.map(str::trim)
.filter(|value| !value.is_empty())
else {
return Ok(false);
};
if !candidate.allow_sell || candidate.is_paused {
return Ok(false);
}
if !self.can_sell_position(ctx, date, symbol) {
@@ -7860,6 +7868,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -8000,6 +8009,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -8145,6 +8155,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -8240,6 +8251,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -8352,6 +8364,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -8469,6 +8482,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -8619,6 +8633,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
},
CandidateEligibility {
date: second_date,
@@ -8630,6 +8645,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
},
],
vec![
@@ -8900,6 +8916,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
},
CandidateEligibility {
date,
@@ -8911,6 +8928,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
},
],
vec![BenchmarkSnapshot {
@@ -9189,6 +9207,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
},
CandidateEligibility {
date,
@@ -9200,6 +9219,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
},
],
vec![BenchmarkSnapshot {
@@ -9418,6 +9438,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
},
CandidateEligibility {
date: second_date,
@@ -9429,6 +9450,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
},
],
vec![
@@ -9626,6 +9648,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -9743,6 +9766,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -9880,6 +9904,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -9985,6 +10010,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -10118,6 +10144,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -10219,6 +10246,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -10348,6 +10376,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -10481,6 +10510,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
benchmark_rows,
)
@@ -10588,6 +10618,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date: current,
@@ -10699,6 +10730,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -10812,6 +10844,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -10868,7 +10901,7 @@ mod tests {
}
#[test]
fn platform_aiquant_firisk_buy_disabled_forces_existing_position_exit() {
fn platform_aiquant_firisk_risk_level_forces_existing_position_exit() {
let date = d(2023, 5, 5);
let symbol = "300621.SZ";
let data = DataSet::from_components(
@@ -10924,6 +10957,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: Some("A".to_string()),
}],
vec![BenchmarkSnapshot {
date,
@@ -11067,6 +11101,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -11221,6 +11256,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -11374,6 +11410,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -11527,6 +11564,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -11693,6 +11731,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -11850,6 +11889,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
[first_date, second_date]
@@ -11997,6 +12037,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -12135,6 +12176,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -12254,6 +12296,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -12403,6 +12446,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -12499,6 +12543,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -12721,6 +12766,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
})
.collect(),
@@ -12911,6 +12957,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -13009,6 +13056,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -13123,6 +13171,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -13224,6 +13273,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -13405,6 +13455,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -13613,6 +13664,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -13781,6 +13833,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -13922,6 +13975,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -14080,6 +14134,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
},
CandidateEligibility {
date: execution_date,
@@ -14091,6 +14146,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
},
],
vec![
@@ -14233,6 +14289,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
},
CandidateEligibility {
date,
@@ -14244,6 +14301,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
},
],
vec![BenchmarkSnapshot {
@@ -14363,6 +14421,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect::<Vec<_>>();
let benchmark_rows = dates
@@ -14505,6 +14564,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect::<Vec<_>>();
let benchmark_rows = dates
@@ -14652,6 +14712,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -14786,6 +14847,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -14921,6 +14983,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -15077,6 +15140,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -15211,6 +15275,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -15367,6 +15432,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -15526,6 +15592,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -15674,6 +15741,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -15847,6 +15915,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
})
.collect(),
@@ -16084,6 +16153,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
})
.collect(),
@@ -16243,6 +16313,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -16387,6 +16458,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -16567,6 +16639,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -16744,6 +16817,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -16965,6 +17039,7 @@ mod tests {
allow_sell: *symbol != "000001.SZ",
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -17114,6 +17189,7 @@ mod tests {
allow_sell: *symbol != "000001.SZ",
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -17265,6 +17341,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -17484,6 +17561,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -17628,6 +17706,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -17766,6 +17845,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -17901,6 +17981,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
@@ -18029,6 +18110,7 @@ mod tests {
allow_sell: false,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -18130,6 +18212,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -18282,6 +18365,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
})
.collect(),
@@ -18453,6 +18537,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
})
.collect(),
@@ -18601,6 +18686,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -18731,6 +18817,7 @@ mod tests {
allow_buy: true,
allow_sell: true,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -18850,6 +18937,7 @@ mod tests {
allow_buy: true,
allow_sell: true,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -19303,6 +19391,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -19426,6 +19515,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -19540,6 +19630,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -19662,6 +19753,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -19802,6 +19894,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -19929,6 +20022,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -20051,6 +20145,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -20167,6 +20262,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -20305,6 +20401,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
dates
@@ -20488,6 +20585,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
})
.collect(),
vec![BenchmarkSnapshot {
crates/fidc-core/src/portfolio.rs
+2
View File
@@ -1086,6 +1086,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
@@ -1172,6 +1173,7 @@ mod tests {
allow_sell: true,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}],
vec![BenchmarkSnapshot {
date,
crates/fidc-core/src/risk_control.rs
+1
View File
@@ -155,6 +155,7 @@ mod tests {
allow_sell: false,
is_kcb: false,
is_one_yuan: false,
risk_level_code: None,
}
}