Merge pull request #64 from lambdaclass/fill-na-balance

update_balance now fills nan values with zero when computing options …
This commit is contained in:
Juan Pablo Amoroso
2020-03-19 13:43:30 -03:00
committed by GitHub
+2 -2
View File
@@ -283,9 +283,9 @@ class Backtest:
current[cost_field] = -current[cost_field]
if (leg_inventory_contract['type'] == Type.CALL.value).any():
calls_value += current[cost_field] * qty * self.shares_per_contract
calls_value = calls_value.add(current[cost_field] * qty * self.shares_per_contract, fill_value=0)
else:
puts_value += current[cost_field] * qty * self.shares_per_contract
puts_value = puts_value.add(current[cost_field] * qty * self.shares_per_contract, fill_value=0)
stocks_current = self._stocks_inventory[['symbol', 'qty']].merge(stocks_data[['date', 'symbol', 'adjClose']],
on='symbol')