Colton Sellers
20910ca2dc
* Remove regression references to non-existant Python versions * Adjust regressions estimated capacity not adjusted by #5389 * Adjusts regression algorithms so that they pass (Index/Index Options) * Changes start/end date on BasicTemplateIndexAlgorithm * Changes option pricing model to BlackScholes in IndexOptionCallITMGreeksExpiryRegressionAlgorithm - The root cause of why there are no greeks at times for these options was identified. It is most likely due to the underlying's VolatilityModel not having had enough data to be "warmed up", which means it will return a standard deviation of zero to the option pricing model, rendering most metrics as NaN. * Adds missing index/index options regression algorithms - Regression algorithms are now 1-1 between C# and Python for Indexes/Index options. All regression tests are now passing * Fixes broken BasicTemplateIndex regression algorithm * Previously traded SPY, but because we have no SPY data in Lean master, I instead opted for index options, since data for those dates is already included * Deal with weekend for breaking test case * Adjust DefaultEndDate test to always pass * Check todays date for open Co-authored-by: Gerardo Salazar <gsalaz9800@gmail.com>
111 lines
5.5 KiB
Python
111 lines
5.5 KiB
Python
# QUANTCONNECT.COM - Democratizing Finance, Empowering Individuals.
|
|
# Lean Algorithmic Trading Engine v2.0. Copyright 2014 QuantConnect Corporation.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at http:#www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License
|
|
|
|
from datetime import datetime, timedelta
|
|
from QuantConnect.Algorithm import *
|
|
from QuantConnect.Data import *
|
|
from QuantConnect.Data.Market import *
|
|
from QuantConnect.Orders import *
|
|
from QuantConnect.Securities import *
|
|
from QuantConnect.Securities.Future import *
|
|
from QuantConnect import Market
|
|
from QuantConnect import *
|
|
|
|
|
|
### <summary>
|
|
### This regression algorithm tests In The Money (ITM) index option expiry for short calls.
|
|
### We expect 2 orders from the algorithm, which are:
|
|
###
|
|
### * Initial entry, sell SPX Call Option (expiring ITM)
|
|
### * Option assignment
|
|
###
|
|
### Additionally, we test delistings for index options and assert that our
|
|
### portfolio holdings reflect the orders the algorithm has submitted.
|
|
### </summary>
|
|
class IndexOptionShortCallITMExpiryRegressionAlgorithm(QCAlgorithm):
|
|
|
|
def Initialize(self):
|
|
self.SetStartDate(2021, 1, 4)
|
|
self.SetEndDate(2021, 1, 31)
|
|
|
|
self.spx = self.AddIndex("SPX", Resolution.Minute).Symbol
|
|
|
|
# Select a index option expiring ITM, and adds it to the algorithm.
|
|
self.spxOption = list(self.OptionChainProvider.GetOptionContractList(self.spx, self.Time))
|
|
self.spxOption = [i for i in self.spxOption if i.ID.StrikePrice <= 3200 and i.ID.OptionRight == OptionRight.Call and i.ID.Date.year == 2021 and i.ID.Date.month == 1]
|
|
self.spxOption = list(sorted(self.spxOption, key=lambda x: x.ID.StrikePrice, reverse=True))[0]
|
|
self.spxOption = self.AddIndexOptionContract(self.spxOption, Resolution.Minute).Symbol
|
|
|
|
self.expectedContract = Symbol.CreateOption(self.spx, Market.USA, OptionStyle.European, OptionRight.Call, 3200, datetime(2021, 1, 15))
|
|
if self.spxOption != self.expectedContract:
|
|
raise Exception(f"Contract self.expectedContract was not found in the chain")
|
|
|
|
self.Schedule.On(self.DateRules.Tomorrow, self.TimeRules.AfterMarketOpen(self.spx, 1), lambda: self.MarketOrder(self.spxOption, -1))
|
|
|
|
def OnData(self, data: Slice):
|
|
# Assert delistings, so that we can make sure that we receive the delisting warnings at
|
|
# the expected time. These assertions detect bug #4872
|
|
for delisting in data.Delistings.Values:
|
|
if delisting.Type == DelistingType.Warning:
|
|
if delisting.Time != datetime(2021, 1, 15):
|
|
raise Exception(f"Delisting warning issued at unexpected date: {delisting.Time}")
|
|
|
|
if delisting.Type == DelistingType.Delisted:
|
|
if delisting.Time != datetime(2021, 1, 16):
|
|
raise Exception(f"Delisting happened at unexpected date: {delisting.Time}")
|
|
|
|
|
|
|
|
|
|
|
|
def OnOrderEvent(self, orderEvent: OrderEvent):
|
|
if orderEvent.Status != OrderStatus.Filled:
|
|
# There's lots of noise with OnOrderEvent, but we're only interested in fills.
|
|
return
|
|
|
|
|
|
if orderEvent.Symbol not in self.Securities:
|
|
raise Exception(f"Order event Symbol not found in Securities collection: {orderEvent.Symbol}")
|
|
|
|
security = self.Securities[orderEvent.Symbol]
|
|
if security.Symbol == self.spx:
|
|
self.AssertIndexOptionOrderExercise(orderEvent, security, self.Securities[self.expectedContract])
|
|
elif security.Symbol == self.expectedContract:
|
|
self.AssertIndexOptionContractOrder(orderEvent, security)
|
|
else:
|
|
raise Exception(f"Received order event for unknown Symbol: {orderEvent.Symbol}")
|
|
|
|
def AssertIndexOptionOrderExercise(self, orderEvent: OrderEvent, index: Security, optionContract: Security):
|
|
if "Assignment" in orderEvent.Message:
|
|
if orderEvent.FillPrice != 3200:
|
|
raise Exception("Option was not assigned at expected strike price (3200)")
|
|
|
|
if orderEvent.Direction != OrderDirection.Sell or index.Holdings.Quantity != 0:
|
|
raise Exception(f"Expected Qty: 0 index holdings for assigned index option {index.Symbol}, found {index.Holdings.Quantity}")
|
|
|
|
def AssertIndexOptionContractOrder(self, orderEvent: OrderEvent, option: Security):
|
|
if orderEvent.Direction == OrderDirection.Sell and option.Holdings.Quantity != -1:
|
|
raise Exception(f"No holdings were created for option contract {option.Symbol}")
|
|
|
|
if orderEvent.IsAssignment and option.Holdings.Quantity != 0:
|
|
raise Exception(f"Holdings were found after option contract was assigned: {option.Symbol}")
|
|
|
|
### <summary>
|
|
### Ran at the end of the algorithm to ensure the algorithm has no holdings
|
|
### </summary>
|
|
### <exception cref="Exception">The algorithm has holdings</exception>
|
|
def OnEndOfAlgorithm(self):
|
|
if self.Portfolio.Invested:
|
|
raise Exception(f"Expected no holdings at end of algorithm, but are invested in: {', '.join(self.Portfolio.Keys)}")
|
|
|
|
|