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>
99 lines
5.1 KiB
Python
99 lines
5.1 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 Out of The Money (OTM) index option expiry for short calls.
|
|
### We expect 2 orders from the algorithm, which are:
|
|
###
|
|
### * Initial entry, sell SPX Call Option (expiring OTM)
|
|
### - Profit the option premium, since the option was not assigned.
|
|
###
|
|
### * Liquidation of SPX call OTM contract on the last trade date
|
|
###
|
|
### Additionally, we test delistings for index options and assert that our
|
|
### portfolio holdings reflect the orders the algorithm has submitted.
|
|
### </summary>
|
|
class IndexOptionShortCallOTMExpiryRegressionAlgorithm(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 >= 4250 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))[0]
|
|
self.spxOption = self.AddIndexOptionContract(self.spxOption, Resolution.Minute).Symbol
|
|
|
|
self.expectedContract = Symbol.CreateOption(self.spx, Market.USA, OptionStyle.European, OptionRight.Call, 4250, 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:
|
|
raise Exception(f"Expected no order events for underlying Symbol {security.Symbol}")
|
|
|
|
if security.Symbol == self.expectedContract:
|
|
self.AssertIndexOptionContractOrder(orderEvent, security)
|
|
else:
|
|
raise Exception(f"Received order event for unknown Symbol: {orderEvent.Symbol}")
|
|
|
|
def AssertIndexOptionContractOrder(self, orderEvent: OrderEvent, optionContract: Security):
|
|
if orderEvent.Direction == OrderDirection.Sell and optionContract.Holdings.Quantity != -1:
|
|
raise Exception(f"No holdings were created for option contract {optionContract.Symbol}")
|
|
if orderEvent.Direction == OrderDirection.Buy and optionContract.Holdings.Quantity != 0:
|
|
raise Exception("Expected no options holdings after closing position")
|
|
if orderEvent.IsAssignment:
|
|
raise Exception(f"Assignment was not expected for {orderEvent.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)}") |