mirror of
https://github.com/wassname/catalyst.git
synced 2026-09-12 12:12:04 +08:00
ENH: Add max leverage account guard
This commit is contained in:
@@ -47,11 +47,12 @@ class TestMinuteRisk(unittest.TestCase):
|
||||
|
||||
first_dt = self.sim_params.first_open
|
||||
second_dt = self.sim_params.first_open + datetime.timedelta(minutes=1)
|
||||
account = {'leverage': 0.0}
|
||||
|
||||
risk_metrics.update(first_dt, 1.0, 2.0)
|
||||
risk_metrics.update(first_dt, 1.0, 2.0, account)
|
||||
|
||||
self.assertEquals(1, len(risk_metrics.metrics.alpha.valid()))
|
||||
|
||||
risk_metrics.update(second_dt, 3.0, 4.0)
|
||||
risk_metrics.update(second_dt, 3.0, 4.0, account)
|
||||
|
||||
self.assertEquals(2, len(risk_metrics.metrics.alpha.valid()))
|
||||
|
||||
@@ -56,7 +56,8 @@ class TestRisk(unittest.TestCase):
|
||||
for dt, returns in answer_key.RETURNS_DATA.iterrows():
|
||||
self.cumulative_metrics_06.update(dt,
|
||||
returns['Algorithm Returns'],
|
||||
returns['Benchmark Returns'])
|
||||
returns['Benchmark Returns'],
|
||||
{'leverage': 0.0})
|
||||
|
||||
def test_algorithm_volatility_06(self):
|
||||
algo_vol_answers = answer_key.RISK_CUMULATIVE.volatility
|
||||
|
||||
Reference in New Issue
Block a user