github-featured/quantconnect--lean
2
0
Fork 0
Code Issues Pull Requests Deployments Models Agent Notes Packages Projects Releases Wiki Activity

Address peer review

Browse Source
This commit is contained in:
Jhonathan Abreu
2023-06-16 17:39:28 -04:00
parent 902ec2c80d
commit 865a733cbf
5 changed files with 29 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Algorithm.Python/StatisticsResultsAlgorithm.py
+9
View File
@@ -89,6 +89,15 @@ class StatisticsResultsAlgorithm(QCAlgorithm):
# Let's keep track of our custom summary statistics after the update
self.CheckMostTradedSecurityStatistic(statistics, most_trade_security, most_trade_security_trade_count)
def OnEndOfAlgorithm(self):
statistics = self.Statistics.Summary
if StatisticsResultsAlgorithm.MostTradedSecurityStatistic not in statistics:
raise Exception(f"Statistic {StatisticsResultsAlgorithm.MostTradedSecurityStatistic} should be in the summary statistics")
if StatisticsResultsAlgorithm.MostTradedSecurityTradeCountStatistic not in statistics:
raise Exception(f"Statistic {StatisticsResultsAlgorithm.MostTradedSecurityTradeCountStatistic} should be in the summary statistics")
most_trade_security, most_trade_security_trade_count = self.GetMostTradeSecurity()
self.CheckMostTradedSecurityStatistic(statistics, most_trade_security, most_trade_security_trade_count)
def CheckMostTradedSecurityStatistic(self, statistics: Dict[str, str], mostTradedSecurity: Symbol, tradeCount: int):
mostTradedSecurityStatistic = statistics[StatisticsResultsAlgorithm.MostTradedSecurityStatistic]