chore: import upstream snapshot with attribution
This commit is contained in:
wehub-resource-sync
@@ -0,0 +1,764 @@
|
||||
/*
|
||||
* 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.
|
||||
*
|
||||
*/
|
||||
|
||||
using NUnit.Framework;
|
||||
using Python.Runtime;
|
||||
using QuantConnect.Algorithm;
|
||||
using QuantConnect.Brokerages;
|
||||
using QuantConnect.Data.Shortable;
|
||||
using QuantConnect.Data.UniverseSelection;
|
||||
using QuantConnect.Interfaces;
|
||||
using QuantConnect.Lean.Engine.DataFeeds;
|
||||
using QuantConnect.Orders;
|
||||
using QuantConnect.Orders.Fees;
|
||||
using QuantConnect.Orders.Fills;
|
||||
using QuantConnect.Orders.Slippage;
|
||||
using QuantConnect.Python;
|
||||
using QuantConnect.Securities;
|
||||
using QuantConnect.Securities.Cfd;
|
||||
using QuantConnect.Securities.Crypto;
|
||||
using QuantConnect.Securities.CryptoFuture;
|
||||
using QuantConnect.Securities.Equity;
|
||||
using QuantConnect.Securities.Forex;
|
||||
using QuantConnect.Securities.Future;
|
||||
using QuantConnect.Securities.IndexOption;
|
||||
using QuantConnect.Securities.Option;
|
||||
using QuantConnect.Tests.Engine.DataFeeds;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.IO;
|
||||
using System.Linq;
|
||||
using Index = QuantConnect.Securities.Index.Index;
|
||||
|
||||
namespace QuantConnect.Tests.Algorithm
|
||||
{
|
||||
[TestFixture]
|
||||
public class AlgorithmAddSecurityTests
|
||||
{
|
||||
private QCAlgorithm _algo;
|
||||
private NullDataFeed _dataFeed;
|
||||
|
||||
/// <summary>
|
||||
/// Instatiate a new algorithm before each test.
|
||||
/// Clear the <see cref="SymbolCache"/> so that no symbols and associated brokerage models are cached between test
|
||||
/// </summary>
|
||||
[SetUp]
|
||||
public void Setup()
|
||||
{
|
||||
_algo = new QCAlgorithm();
|
||||
_dataFeed = new NullDataFeed
|
||||
{
|
||||
ShouldThrow = false
|
||||
};
|
||||
_algo.SubscriptionManager.SetDataManager(new DataManagerStub(_dataFeed, _algo));
|
||||
}
|
||||
|
||||
[Test, TestCaseSource(nameof(TestAddSecurityWithSymbol))]
|
||||
public void AddSecurityWithSymbol(Symbol symbol, Type type = null)
|
||||
{
|
||||
var security = type != null ? _algo.AddData(type, symbol.Underlying) : _algo.AddSecurity(symbol);
|
||||
Assert.AreEqual(security.Symbol, symbol);
|
||||
Assert.AreEqual((Symbol)security, symbol);
|
||||
Assert.IsTrue(_algo.Securities.ContainsKey(symbol));
|
||||
|
||||
Assert.DoesNotThrow(() =>
|
||||
{
|
||||
switch (symbol.SecurityType)
|
||||
{
|
||||
case SecurityType.Equity:
|
||||
var equity = (Equity)security;
|
||||
break;
|
||||
case SecurityType.Option:
|
||||
var option = (Option)security;
|
||||
break;
|
||||
case SecurityType.Forex:
|
||||
var forex = (Forex)security;
|
||||
break;
|
||||
case SecurityType.Future:
|
||||
var future = (Future)security;
|
||||
break;
|
||||
case SecurityType.Cfd:
|
||||
var cfd = (Cfd)security;
|
||||
break;
|
||||
case SecurityType.Index:
|
||||
var index = (Index)security;
|
||||
break;
|
||||
case SecurityType.IndexOption:
|
||||
var indexOption = (IndexOption)security;
|
||||
break;
|
||||
case SecurityType.Crypto:
|
||||
var crypto = (Crypto)security;
|
||||
break;
|
||||
case SecurityType.CryptoFuture:
|
||||
var cryptoFuture = (CryptoFuture)security;
|
||||
break;
|
||||
case SecurityType.Base:
|
||||
break;
|
||||
default:
|
||||
throw new Exception($"Invalid Security Type: {symbol.SecurityType}");
|
||||
}
|
||||
});
|
||||
|
||||
if (symbol.IsCanonical())
|
||||
{
|
||||
Assert.DoesNotThrow(() => _algo.OnEndOfTimeStep());
|
||||
|
||||
Assert.IsTrue(_algo.UniverseManager.ContainsKey(symbol));
|
||||
}
|
||||
}
|
||||
|
||||
[TestCaseSource(nameof(GetDataNormalizationModes))]
|
||||
public void AddsEquityWithExpectedDataNormalizationMode(DataNormalizationMode dataNormalizationMode)
|
||||
{
|
||||
var equity = _algo.AddEquity("AAPL", dataNormalizationMode: dataNormalizationMode);
|
||||
Assert.That(_algo.SubscriptionManager.Subscriptions.Where(x => x.Symbol == equity.Symbol).Select(x => x.DataNormalizationMode),
|
||||
Has.All.EqualTo(dataNormalizationMode));
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void ProperlyAddsFutureWithExtendedMarketHours(
|
||||
[Values(true, false)] bool extendedMarketHours,
|
||||
[ValueSource(nameof(FuturesTestCases))] Func<QCAlgorithm, Security> getFuture)
|
||||
{
|
||||
var future = _algo.AddFuture(Futures.Indices.VIX, Resolution.Minute, extendedMarketHours: extendedMarketHours);
|
||||
var subscriptions = _algo.SubscriptionManager.Subscriptions.Where(x => x.Symbol == future.Symbol).ToList();
|
||||
|
||||
var universeSubscriptions = subscriptions.Where(x => x.Type == typeof(FutureUniverse)).ToList();
|
||||
Assert.AreEqual(1, universeSubscriptions.Count);
|
||||
// Universe does not support extended market hours
|
||||
Assert.IsFalse(universeSubscriptions[0].ExtendedMarketHours);
|
||||
|
||||
var nonUniverseSubscriptions = subscriptions.Where(x => x.Type != typeof(FutureUniverse)).ToList();
|
||||
Assert.Greater(nonUniverseSubscriptions.Count, 0);
|
||||
Assert.That(nonUniverseSubscriptions.Select(x => x.ExtendedMarketHours),
|
||||
Has.All.EqualTo(extendedMarketHours));
|
||||
}
|
||||
|
||||
[TestCaseSource(nameof(FuturesTestCases))]
|
||||
public void AddFutureWithExtendedMarketHours(Func<QCAlgorithm, Security> getFuture)
|
||||
{
|
||||
string file = Path.Combine("TestData", "SampleMarketHoursDatabase.json");
|
||||
var marketHoursDatabase = MarketHoursDatabase.FromFile(file);
|
||||
var securityService = new SecurityService(
|
||||
_algo.Portfolio.CashBook,
|
||||
marketHoursDatabase,
|
||||
SymbolPropertiesDatabase.FromDataFolder(),
|
||||
_algo,
|
||||
RegisteredSecurityDataTypesProvider.Null,
|
||||
new SecurityCacheProvider(_algo.Portfolio),
|
||||
algorithm: _algo);
|
||||
_algo.Securities.SetSecurityService(securityService);
|
||||
|
||||
var future = getFuture(_algo);
|
||||
|
||||
var now = new DateTime(2022, 6, 26, 17, 0, 0);
|
||||
Assert.AreEqual(DayOfWeek.Sunday, now.DayOfWeek);
|
||||
var regularMarketStartTime = new TimeSpan(8, 30, 0);
|
||||
var regularMarketEndTime = new TimeSpan(15, 0, 0);
|
||||
var firstExtendedMarketStartTime = regularMarketEndTime;
|
||||
var firstExtendedMarketEndTime = new TimeSpan(16, 0, 0);
|
||||
var secondExtendedMarketStartTime = new TimeSpan(17, 0, 0);
|
||||
|
||||
Action<DateTime> checkExtendedHours = (date) =>
|
||||
{
|
||||
Assert.IsFalse(future.Exchange.Hours.IsOpen(now, false));
|
||||
Assert.IsTrue(future.Exchange.Hours.IsOpen(now, true));
|
||||
};
|
||||
Action<DateTime> checkRegularHours = (date) =>
|
||||
{
|
||||
Assert.IsTrue(future.Exchange.Hours.IsOpen(now, false));
|
||||
Assert.IsTrue(future.Exchange.Hours.IsOpen(now, true));
|
||||
};
|
||||
Action<DateTime> checkClosed = (date) =>
|
||||
{
|
||||
Assert.IsFalse(future.Exchange.Hours.IsOpen(now, false));
|
||||
Assert.IsFalse(future.Exchange.Hours.IsOpen(now, true));
|
||||
};
|
||||
|
||||
while (now.DayOfWeek < DayOfWeek.Saturday)
|
||||
{
|
||||
while (now.TimeOfDay < regularMarketStartTime)
|
||||
{
|
||||
checkExtendedHours(now);
|
||||
now = now.AddMinutes(1);
|
||||
}
|
||||
|
||||
while (now.TimeOfDay < regularMarketEndTime)
|
||||
{
|
||||
checkRegularHours(now);
|
||||
now = now.AddMinutes(1);
|
||||
}
|
||||
|
||||
while (now.TimeOfDay >= firstExtendedMarketStartTime && now.TimeOfDay < firstExtendedMarketEndTime)
|
||||
{
|
||||
checkExtendedHours(now);
|
||||
now = now.AddMinutes(1);
|
||||
}
|
||||
|
||||
while (now.TimeOfDay < secondExtendedMarketStartTime)
|
||||
{
|
||||
checkClosed(now);
|
||||
now = now.AddMinutes(1);
|
||||
}
|
||||
|
||||
var endOfDay = now.AddDays(1).Date;
|
||||
if (now.DayOfWeek < DayOfWeek.Friday)
|
||||
{
|
||||
while (now < endOfDay)
|
||||
{
|
||||
checkExtendedHours(now);
|
||||
now = now.AddMinutes(1);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
now = endOfDay;
|
||||
}
|
||||
}
|
||||
|
||||
while (now.DayOfWeek < DayOfWeek.Sunday)
|
||||
{
|
||||
checkClosed(now);
|
||||
now = now.AddMinutes(1);
|
||||
}
|
||||
}
|
||||
|
||||
// Reproduces https://github.com/QuantConnect/Lean/issues/7451
|
||||
[Test]
|
||||
public void DoesNotAddExtraIndexSubscriptionAfterAddingIndexOptionContract()
|
||||
{
|
||||
var spx = _algo.AddIndex("SPX", Resolution.Minute, fillForward: false);
|
||||
|
||||
Assert.AreEqual(1, _algo.SubscriptionManager.Subscriptions.Count());
|
||||
Assert.AreEqual(spx.Symbol, _algo.SubscriptionManager.Subscriptions.Single().Symbol);
|
||||
|
||||
var spxOption = Symbol.CreateOption(
|
||||
spx.Symbol,
|
||||
Market.USA,
|
||||
OptionStyle.European,
|
||||
OptionRight.Call,
|
||||
3200m,
|
||||
new DateTime(2021, 1, 15));
|
||||
_algo.AddIndexOptionContract(spxOption, Resolution.Minute);
|
||||
|
||||
Assert.Greater(_algo.SubscriptionManager.Subscriptions.Count(), 1);
|
||||
Assert.AreEqual(1, _algo.SubscriptionManager.Subscriptions.Count(x => x.Symbol == spx.Symbol));
|
||||
}
|
||||
|
||||
[TestCase("SPXW", "SPX")]
|
||||
[TestCase("RUTW", "RUT")]
|
||||
[TestCase("VIXW", "VIX")]
|
||||
[TestCase("NDXP", "NDX")]
|
||||
[TestCase("NQX", "NDX")]
|
||||
[TestCase("SPX", "SPX")]
|
||||
[TestCase("VIX", "VIX")]
|
||||
public void AddIndexOptionMapsNonStandardOptionTickerToItsUnderlyingIndex(string ticker, string expectedUnderlyingTicker)
|
||||
{
|
||||
var option = _algo.AddIndexOption(ticker);
|
||||
|
||||
// the canonical keeps the provided option ticker
|
||||
Assert.AreEqual($"?{ticker}", option.Symbol.Value);
|
||||
Assert.AreEqual(ticker, option.Symbol.ID.Symbol);
|
||||
// but the underlying is the actual index, which has data. Else the option contracts would reference
|
||||
// a data-less underlying index security which would never get a price
|
||||
Assert.AreEqual(SecurityType.Index, option.Symbol.Underlying.SecurityType);
|
||||
Assert.AreEqual(expectedUnderlyingTicker, option.Symbol.Underlying.Value);
|
||||
|
||||
// the auto-added underlying index security is the mapped index
|
||||
_algo.OnEndOfTimeStep();
|
||||
Assert.IsTrue(_algo.Securities.ContainsKey(option.Symbol.Underlying));
|
||||
Assert.AreEqual(option.Symbol.Underlying, option.Underlying.Symbol);
|
||||
if (ticker != expectedUnderlyingTicker)
|
||||
{
|
||||
Assert.IsFalse(_algo.Securities.Keys.Any(x => x.SecurityType == SecurityType.Index && x.Value == ticker));
|
||||
// a one time warning is sent about the mapping
|
||||
Assert.AreEqual(1, _algo.DebugMessages.Count(x => x.Contains(ticker, StringComparison.InvariantCulture)));
|
||||
}
|
||||
else
|
||||
{
|
||||
Assert.IsEmpty(_algo.DebugMessages);
|
||||
}
|
||||
}
|
||||
|
||||
[TestCase(Language.CSharp)]
|
||||
[TestCase(Language.Python)]
|
||||
public void AddSecurityInitializerAppendsInitializer(Language language)
|
||||
{
|
||||
var algorithm = new AlgorithmStub();
|
||||
|
||||
Assert.IsNotAssignableFrom<CompositeSecurityInitializer>(algorithm.SecurityInitializer);
|
||||
|
||||
if (language == Language.CSharp)
|
||||
{
|
||||
var classInitializer1 = new TestCustomSecurityInitializer();
|
||||
algorithm.AddSecurityInitializer(classInitializer1);
|
||||
|
||||
Assert.IsInstanceOf<CompositeSecurityInitializer>(algorithm.SecurityInitializer);
|
||||
|
||||
var classInitializer2 = new TestCustomSecurityInitializer();
|
||||
algorithm.AddSecurityInitializer(classInitializer2);
|
||||
|
||||
var funcInitializer1CallCount = 0;
|
||||
algorithm.AddSecurityInitializer((_) => funcInitializer1CallCount++);
|
||||
|
||||
var funcInitializer2CallCount = 0;
|
||||
algorithm.AddSecurityInitializer((_) => funcInitializer2CallCount++);
|
||||
|
||||
var security = algorithm.AddEquity("SPY");
|
||||
Assert.AreEqual(1, classInitializer1.CallCount);
|
||||
Assert.AreEqual(1, classInitializer2.CallCount);
|
||||
Assert.AreEqual(1, funcInitializer1CallCount);
|
||||
Assert.AreEqual(1, funcInitializer2CallCount);
|
||||
}
|
||||
else
|
||||
{
|
||||
using var _ = Py.GIL();
|
||||
using var module = PyModule.FromString("AddSecurityInitializerAppendsInitializer", @"
|
||||
class TestCustomSecurityInitializer:
|
||||
def __init__(self):
|
||||
self.call_count = 0
|
||||
|
||||
def initialize(self, security):
|
||||
self.call_count += 1
|
||||
|
||||
class_initializer1 = TestCustomSecurityInitializer()
|
||||
class_initializer2 = TestCustomSecurityInitializer()
|
||||
func_call_count1 = 0
|
||||
func_call_count2 = 0
|
||||
|
||||
def add_security_initializers(algorithm):
|
||||
algorithm.add_security_initializer(class_initializer1)
|
||||
algorithm.add_security_initializer(class_initializer2)
|
||||
|
||||
algorithm.add_security_initializer(func_security_initializer1)
|
||||
algorithm.add_security_initializer(func_security_initializer2)
|
||||
|
||||
def func_security_initializer1(security):
|
||||
global func_call_count1
|
||||
func_call_count1 += 1
|
||||
|
||||
def func_security_initializer2(security):
|
||||
global func_call_count2
|
||||
func_call_count2 += 1
|
||||
");
|
||||
|
||||
using var addSecurityInitializers = module.GetAttr("add_security_initializers");
|
||||
using var pyAlgorithm = algorithm.ToPython();
|
||||
addSecurityInitializers.Invoke(pyAlgorithm);
|
||||
|
||||
Assert.IsInstanceOf<CompositeSecurityInitializer>(algorithm.SecurityInitializer);
|
||||
|
||||
var security = algorithm.AddEquity("SPY");
|
||||
|
||||
using var classInitializer1 = module.GetAttr("class_initializer1");
|
||||
var classInitializer1CallCount = classInitializer1.GetAttr("call_count").GetAndDispose<int>();
|
||||
Assert.AreEqual(1, classInitializer1CallCount);
|
||||
|
||||
using var classInitializer2 = module.GetAttr("class_initializer2");
|
||||
var classInitializer2CallCount = classInitializer2.GetAttr("call_count").GetAndDispose<int>();
|
||||
Assert.AreEqual(1, classInitializer2CallCount);
|
||||
|
||||
var funcCallCount1 = module.GetAttr("func_call_count1").GetAndDispose<int>();
|
||||
Assert.AreEqual(1, funcCallCount1);
|
||||
|
||||
var funcCallCount2 = module.GetAttr("func_call_count2").GetAndDispose<int>();
|
||||
Assert.AreEqual(1, funcCallCount2);
|
||||
}
|
||||
}
|
||||
|
||||
[TestCase(Language.CSharp)]
|
||||
[TestCase(Language.Python)]
|
||||
public void SetSecurityInitializerReplacesInitializer(Language language)
|
||||
{
|
||||
var algorithm = new AlgorithmStub();
|
||||
|
||||
Assert.IsNotAssignableFrom<CompositeSecurityInitializer>(algorithm.SecurityInitializer);
|
||||
|
||||
if (language == Language.CSharp)
|
||||
{
|
||||
var classInitializer1 = new TestCustomSecurityInitializer();
|
||||
algorithm.AddSecurityInitializer(classInitializer1);
|
||||
|
||||
Assert.IsInstanceOf<CompositeSecurityInitializer>(algorithm.SecurityInitializer);
|
||||
|
||||
algorithm.SetSecurityInitializer(classInitializer1);
|
||||
Assert.IsInstanceOf<TestCustomSecurityInitializer>(algorithm.SecurityInitializer);
|
||||
}
|
||||
else
|
||||
{
|
||||
using var _ = Py.GIL();
|
||||
using var module = PyModule.FromString("AddSecurityInitializerAppendsInitializer", @"
|
||||
class TestCustomSecurityInitializer:
|
||||
def initialize(self, security):
|
||||
pass
|
||||
|
||||
def add_security_initializer(algorithm):
|
||||
algorithm.add_security_initializer(TestCustomSecurityInitializer())
|
||||
|
||||
def set_security_initializer(algorithm):
|
||||
algorithm.set_security_initializer(TestCustomSecurityInitializer())
|
||||
");
|
||||
|
||||
using var pyAlgorithm = algorithm.ToPython();
|
||||
using var addSecurityInitializer = module.GetAttr("add_security_initializer");
|
||||
addSecurityInitializer.Invoke(pyAlgorithm);
|
||||
|
||||
Assert.IsInstanceOf<CompositeSecurityInitializer>(algorithm.SecurityInitializer);
|
||||
|
||||
using var setSecurityInitializer = module.GetAttr("set_security_initializer");
|
||||
setSecurityInitializer.Invoke(pyAlgorithm);
|
||||
|
||||
Assert.IsInstanceOf<SecurityInitializerPythonWrapper>(algorithm.SecurityInitializer);
|
||||
}
|
||||
}
|
||||
|
||||
[TestCase(Language.CSharp)]
|
||||
[TestCase(Language.Python)]
|
||||
public void AddsSecurityInitializerAfterSetting(Language language)
|
||||
{
|
||||
var algorithm = new AlgorithmStub();
|
||||
|
||||
if (language == Language.CSharp)
|
||||
{
|
||||
var classInitializer1 = new TestCustomSecurityInitializer();
|
||||
algorithm.SetSecurityInitializer(classInitializer1);
|
||||
|
||||
Assert.IsAssignableFrom<TestCustomSecurityInitializer>(algorithm.SecurityInitializer);
|
||||
|
||||
var classInitializer2 = new TestCustomSecurityInitializer();
|
||||
algorithm.AddSecurityInitializer(classInitializer2);
|
||||
|
||||
Assert.IsInstanceOf<CompositeSecurityInitializer>(algorithm.SecurityInitializer);
|
||||
|
||||
var funcInitializer1CallCount = 0;
|
||||
algorithm.AddSecurityInitializer((_) => funcInitializer1CallCount++);
|
||||
|
||||
var funcInitializer2CallCount = 0;
|
||||
algorithm.AddSecurityInitializer((_) => funcInitializer2CallCount++);
|
||||
|
||||
var security = algorithm.AddEquity("SPY");
|
||||
Assert.AreEqual(1, classInitializer1.CallCount);
|
||||
Assert.AreEqual(1, classInitializer2.CallCount);
|
||||
Assert.AreEqual(1, funcInitializer1CallCount);
|
||||
Assert.AreEqual(1, funcInitializer2CallCount);
|
||||
}
|
||||
else
|
||||
{
|
||||
using var _ = Py.GIL();
|
||||
using var module = PyModule.FromString("AddSecurityInitializerAppendsInitializer", @"
|
||||
class TestCustomSecurityInitializer:
|
||||
def __init__(self):
|
||||
self.call_count = 0
|
||||
|
||||
def initialize(self, security):
|
||||
self.call_count += 1
|
||||
|
||||
class_initializer1 = TestCustomSecurityInitializer()
|
||||
class_initializer2 = TestCustomSecurityInitializer()
|
||||
func_call_count1 = 0
|
||||
func_call_count2 = 0
|
||||
|
||||
def set_security_initializer(algorithm):
|
||||
algorithm.set_security_initializer(class_initializer1)
|
||||
|
||||
def add_security_initializers(algorithm):
|
||||
algorithm.add_security_initializer(class_initializer2)
|
||||
|
||||
algorithm.add_security_initializer(func_security_initializer1)
|
||||
algorithm.add_security_initializer(func_security_initializer2)
|
||||
|
||||
def func_security_initializer1(security):
|
||||
global func_call_count1
|
||||
func_call_count1 += 1
|
||||
|
||||
def func_security_initializer2(security):
|
||||
global func_call_count2
|
||||
func_call_count2 += 1
|
||||
");
|
||||
|
||||
using var pyAlgorithm = algorithm.ToPython();
|
||||
|
||||
using var setSecurityInitializer = module.GetAttr("set_security_initializer");
|
||||
setSecurityInitializer.Invoke(pyAlgorithm);
|
||||
|
||||
Assert.IsInstanceOf<SecurityInitializerPythonWrapper>(algorithm.SecurityInitializer);
|
||||
|
||||
using var addSecurityInitializers = module.GetAttr("add_security_initializers");
|
||||
addSecurityInitializers.Invoke(pyAlgorithm);
|
||||
|
||||
Assert.IsInstanceOf<CompositeSecurityInitializer>(algorithm.SecurityInitializer);
|
||||
|
||||
var security = algorithm.AddEquity("SPY");
|
||||
|
||||
using var classInitializer1 = module.GetAttr("class_initializer1");
|
||||
var classInitializer1CallCount = classInitializer1.GetAttr("call_count").GetAndDispose<int>();
|
||||
Assert.AreEqual(1, classInitializer1CallCount);
|
||||
|
||||
using var classInitializer2 = module.GetAttr("class_initializer2");
|
||||
var classInitializer2CallCount = classInitializer2.GetAttr("call_count").GetAndDispose<int>();
|
||||
Assert.AreEqual(1, classInitializer2CallCount);
|
||||
|
||||
var funcCallCount1 = module.GetAttr("func_call_count1").GetAndDispose<int>();
|
||||
Assert.AreEqual(1, funcCallCount1);
|
||||
|
||||
var funcCallCount2 = module.GetAttr("func_call_count2").GetAndDispose<int>();
|
||||
Assert.AreEqual(1, funcCallCount2);
|
||||
}
|
||||
}
|
||||
|
||||
[TestCase(Language.CSharp)]
|
||||
[TestCase(Language.Python)]
|
||||
public void SetBrokerageModelAppendsSecurityIntializerAfterAddSecurityInitializer(Language language)
|
||||
{
|
||||
var algorithm = new AlgorithmStub();
|
||||
|
||||
Assert.IsNotAssignableFrom<CompositeSecurityInitializer>(algorithm.SecurityInitializer);
|
||||
|
||||
var brokerageModel = default(TestBrokerageModel);
|
||||
var security = default(Security);
|
||||
|
||||
if (language == Language.CSharp)
|
||||
{
|
||||
var classInitializer = new TestCustomSecurityInitializer();
|
||||
algorithm.AddSecurityInitializer(classInitializer);
|
||||
|
||||
Assert.IsInstanceOf<CompositeSecurityInitializer>(algorithm.SecurityInitializer);
|
||||
|
||||
brokerageModel = new TestBrokerageModel();
|
||||
algorithm.SetBrokerageModel(brokerageModel);
|
||||
|
||||
Assert.AreSame(brokerageModel, algorithm.BrokerageModel);
|
||||
Assert.IsInstanceOf<CompositeSecurityInitializer>(algorithm.SecurityInitializer);
|
||||
|
||||
security = algorithm.AddEquity("SPY");
|
||||
|
||||
Assert.AreEqual(1, classInitializer.CallCount);
|
||||
}
|
||||
else
|
||||
{
|
||||
using var _ = Py.GIL();
|
||||
using var module = PyModule.FromString("SetBrokerageModelAppendsSecurityIntializerAfterAddSecurityInitializer", @"
|
||||
from QuantConnect.Tests.Algorithm import AlgorithmAddSecurityTests
|
||||
|
||||
class TestCustomSecurityInitializer:
|
||||
def __init__(self):
|
||||
self.call_count = 0
|
||||
|
||||
def initialize(self, security):
|
||||
security.set_fill_model(AlgorithmAddSecurityTests.TestCustomSecurityInitializer.TestFillModel())
|
||||
self.call_count += 1
|
||||
|
||||
class_initializer = TestCustomSecurityInitializer()
|
||||
|
||||
def add_security_initializers(algorithm):
|
||||
algorithm.add_security_initializer(class_initializer)
|
||||
|
||||
def set_brokerage_model(algorithm):
|
||||
algorithm.set_brokerage_model(AlgorithmAddSecurityTests.TestBrokerageModel())
|
||||
");
|
||||
|
||||
using var addSecurityInitializers = module.GetAttr("add_security_initializers");
|
||||
using var pyAlgorithm = algorithm.ToPython();
|
||||
addSecurityInitializers.Invoke(pyAlgorithm);
|
||||
|
||||
Assert.IsInstanceOf<CompositeSecurityInitializer>(algorithm.SecurityInitializer);
|
||||
|
||||
using var setBrokerageModel = module.GetAttr("set_brokerage_model");
|
||||
setBrokerageModel.Invoke(pyAlgorithm);
|
||||
|
||||
Assert.IsInstanceOf<CompositeSecurityInitializer>(algorithm.SecurityInitializer);
|
||||
Assert.IsInstanceOf<TestBrokerageModel>(algorithm.BrokerageModel);
|
||||
|
||||
brokerageModel = (TestBrokerageModel)algorithm.BrokerageModel;
|
||||
|
||||
security = algorithm.AddEquity("SPY");
|
||||
|
||||
using var classInitializer = module.GetAttr("class_initializer");
|
||||
var classInitializer1CallCount = classInitializer.GetAttr("call_count").GetAndDispose<int>();
|
||||
Assert.AreEqual(1, classInitializer1CallCount);
|
||||
}
|
||||
|
||||
Assert.IsTrue(brokerageModel.GetFillModelCalled);
|
||||
Assert.IsTrue(brokerageModel.GetFeeModelCalled);
|
||||
Assert.IsTrue(brokerageModel.GetSlippageModelCalled);
|
||||
Assert.IsTrue(brokerageModel.GetSettlementModelCalled);
|
||||
Assert.IsTrue(brokerageModel.GetBuyingPowerModelCalled);
|
||||
Assert.IsTrue(brokerageModel.GetMarginInterestRateModelCalled);
|
||||
Assert.IsTrue(brokerageModel.GetLeverageCalled);
|
||||
Assert.IsTrue(brokerageModel.GetShortableProviderCalled);
|
||||
|
||||
Assert.IsInstanceOf<TestBrokerageModel.TestFeeModel>(security.FeeModel);
|
||||
Assert.IsInstanceOf<TestBrokerageModel.TestSlippageModel>(security.SlippageModel);
|
||||
Assert.IsInstanceOf<TestBrokerageModel.TestSettlementModel>(security.SettlementModel);
|
||||
Assert.IsInstanceOf<TestBrokerageModel.TestBuyingPowerModel>(security.BuyingPowerModel);
|
||||
Assert.IsInstanceOf<TestBrokerageModel.TestMarginInterestRateModel>(security.MarginInterestRateModel);
|
||||
Assert.IsInstanceOf<TestBrokerageModel.TestShortableProvider>(security.ShortableProvider);
|
||||
Assert.AreEqual(5000, security.Leverage);
|
||||
|
||||
// All models should've been set my the TestBrokerageModel, except the fill model,
|
||||
// which should have been set by the TestCustomSecurityInitializer, because user defined
|
||||
// initializer should run after the brokerage model initializer
|
||||
Assert.IsInstanceOf<TestCustomSecurityInitializer.TestFillModel>(security.FillModel);
|
||||
}
|
||||
|
||||
public class TestCustomSecurityInitializer : ISecurityInitializer
|
||||
{
|
||||
public int CallCount { get; private set; }
|
||||
|
||||
public class TestFillModel : FillModel { }
|
||||
|
||||
public void Initialize(Security security)
|
||||
{
|
||||
CallCount++;
|
||||
security.SetFillModel(new TestFillModel());
|
||||
}
|
||||
}
|
||||
|
||||
public class TestBrokerageModel : DefaultBrokerageModel
|
||||
{
|
||||
public bool GetFillModelCalled { get; private set; }
|
||||
public bool GetFeeModelCalled { get; private set; }
|
||||
public bool GetSlippageModelCalled { get; private set; }
|
||||
public bool GetSettlementModelCalled { get; private set; }
|
||||
public bool GetBuyingPowerModelCalled { get; private set; }
|
||||
public bool GetMarginInterestRateModelCalled { get; private set; }
|
||||
public bool GetLeverageCalled { get; private set; }
|
||||
public bool GetShortableProviderCalled { get; private set; }
|
||||
|
||||
public class TestFillModel : FillModel { }
|
||||
public class TestFeeModel : FeeModel { }
|
||||
public class TestSlippageModel : ISlippageModel
|
||||
{
|
||||
public decimal GetSlippageApproximation(Security asset, Order order)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
public class TestSettlementModel : ImmediateSettlementModel { }
|
||||
public class TestBuyingPowerModel : BuyingPowerModel { }
|
||||
public class TestMarginInterestRateModel : IMarginInterestRateModel
|
||||
{
|
||||
public void ApplyMarginInterestRate(MarginInterestRateParameters marginInterestRateParameters)
|
||||
{
|
||||
}
|
||||
}
|
||||
public class TestShortableProvider : NullShortableProvider { }
|
||||
|
||||
public override IFillModel GetFillModel(Security security)
|
||||
{
|
||||
GetFillModelCalled = true;
|
||||
return new TestFillModel();
|
||||
}
|
||||
|
||||
public override IFeeModel GetFeeModel(Security security)
|
||||
{
|
||||
GetFeeModelCalled = true;
|
||||
return new TestFeeModel();
|
||||
}
|
||||
|
||||
public override ISlippageModel GetSlippageModel(Security security)
|
||||
{
|
||||
GetSlippageModelCalled = true;
|
||||
return new TestSlippageModel();
|
||||
}
|
||||
|
||||
public override ISettlementModel GetSettlementModel(Security security)
|
||||
{
|
||||
GetSettlementModelCalled = true;
|
||||
return new TestSettlementModel();
|
||||
}
|
||||
|
||||
public override IBuyingPowerModel GetBuyingPowerModel(Security security)
|
||||
{
|
||||
GetBuyingPowerModelCalled = true;
|
||||
return new TestBuyingPowerModel();
|
||||
}
|
||||
|
||||
public override IMarginInterestRateModel GetMarginInterestRateModel(Security security)
|
||||
{
|
||||
GetMarginInterestRateModelCalled = true;
|
||||
return new TestMarginInterestRateModel();
|
||||
}
|
||||
|
||||
public override decimal GetLeverage(Security security)
|
||||
{
|
||||
GetLeverageCalled = true;
|
||||
return 5000;
|
||||
}
|
||||
|
||||
public override IShortableProvider GetShortableProvider(Security security)
|
||||
{
|
||||
GetShortableProviderCalled = true;
|
||||
return new TestShortableProvider();
|
||||
}
|
||||
}
|
||||
|
||||
private static TestCaseData[] TestAddSecurityWithSymbol
|
||||
{
|
||||
get
|
||||
{
|
||||
var result = new List<TestCaseData>()
|
||||
{
|
||||
new TestCaseData(Symbols.SPY, null),
|
||||
new TestCaseData(Symbols.EURUSD, null),
|
||||
new TestCaseData(Symbols.DE30EUR, null),
|
||||
new TestCaseData(Symbols.BTCUSD, null),
|
||||
new TestCaseData(Symbols.ES_Future_Chain, null),
|
||||
new TestCaseData(Symbols.Future_ESZ18_Dec2018, null),
|
||||
new TestCaseData(Symbols.SPY_Option_Chain, null),
|
||||
new TestCaseData(Symbols.SPY_C_192_Feb19_2016, null),
|
||||
new TestCaseData(Symbols.SPY_P_192_Feb19_2016, null),
|
||||
new TestCaseData(Symbol.Create("CustomData", SecurityType.Base, Market.Binance), null),
|
||||
new TestCaseData(Symbol.Create("CustomData2", SecurityType.Base, Market.COMEX), null)
|
||||
};
|
||||
|
||||
foreach (var market in Market.SupportedMarkets())
|
||||
{
|
||||
foreach (var kvp in SymbolPropertiesDatabase.FromDataFolder().GetSymbolPropertiesList(market))
|
||||
{
|
||||
var securityDatabaseKey = kvp.Key;
|
||||
if (securityDatabaseKey.SecurityType != SecurityType.FutureOption)
|
||||
{
|
||||
result.Add(new TestCaseData(Symbol.Create(securityDatabaseKey.Symbol, securityDatabaseKey.SecurityType,
|
||||
securityDatabaseKey.Market), null));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return result.ToArray();
|
||||
}
|
||||
}
|
||||
|
||||
private static DataNormalizationMode[] GetDataNormalizationModes()
|
||||
{
|
||||
return ((DataNormalizationMode[])Enum.GetValues(typeof(DataNormalizationMode)))
|
||||
.Where(x => x != DataNormalizationMode.ScaledRaw).ToArray();
|
||||
}
|
||||
|
||||
private static Func<QCAlgorithm, Security>[] FuturesTestCases
|
||||
{
|
||||
get
|
||||
{
|
||||
return new Func<QCAlgorithm, Security>[]
|
||||
{
|
||||
(algo) => algo.AddFuture(Futures.Indices.VIX, Resolution.Minute, extendedMarketHours: true),
|
||||
(algo) => algo.AddFutureContract(Symbol.CreateFuture(Futures.Indices.VIX, Market.CFE, new DateTime(2022, 8, 1)),
|
||||
Resolution.Minute, extendedMarketHours: true)
|
||||
};
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user