Files
quantconnect--lean/Tests/Python/PandasIndexingTests.cs
T
Jhonathan Abreu 0a9dc2c71c QCAlgorithm's OptionChain() api refactor (#8334)
* Fix pandas converter to handle list of data with different symbols

* Properly convert list of data into dataframe

Take into consideration data for multiple symbols in the same list

* Cleanup

* Index dataframes by symbol object instead of SID string

* Add symbol equality operator to compare against object

* Exclude "ID" from option chain dataframe

* Minor fix

* Add greeks columns directly in option chain dataframe.

Also add pass-through properties for greek values in OptionUniverse

* Some cleanup

* Minor fix

* Add new QCAlgorithm.OptionChains() method

- Use OptionChains as output
- Add DataFrame to OptionChain and OptionChains
- Rename Greeks classes
- Add ISymbolProvider for classes that have a symbol (IBaseData, OptionContract)

* Unify QCAlgorithmOptionChain API

Also refactor OptionContract to handle: (1) Actual market data and option price model data, and (2) OptionUniverse data

* Pass symbol properties to OptionUniverse option chain from algorithm

* Format OptionContract for dataframe

* Minor fix

* Add multiple option chains api regression algorithms and other minor changes

* Address peer review

Add NullGreeks class: keep ModeledGreeks as internal as possible

* Minor fix and add PandasConverter unit tests

* Peer review: Non-thread-safe Lazy for Python

* Handle Greeks unwrapping by PandasData

* PandasData cleanup

* Add data and other minor changes

* Unit test fix

* Update Pythonnet to 2.0.39

* Cleanup

* PandasData handling children class members

Address peer review

* Fix: indexing symbol conversion in pandas mapper

* Fix pandas mapper to convert string keys to symbol only when necessary

* Cleanup

* Cleanup

* Add PandasColumn python class to handle proper indexing

This allows propery hash and equality between Symbols, C# strings and Python strings

* Minor fixes

* Symbol cache improvements

* Minor fix for cache miss

* Revert PandasMapper reserved names and improvements

* Minor fix

* Revert reserved names

* Minor fix for Symbol equality operators

---------

Co-authored-by: Martin Molinero <martin.molinero1@gmail.com>
2024-10-04 12:26:15 -04:00

90 lines
2.8 KiB
C#

/*
* 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 System;
using NUnit.Framework;
using Python.Runtime;
using static QLNet.NumericHaganPricer;
namespace QuantConnect.Tests.Python
{
[TestFixture]
// TODO: Rename to PandasPythonTests, dedicate class to python tests under ./PandasTests directory
public class PandasIndexingTests
{
private dynamic _module;
private dynamic _pandasIndexingTests;
private dynamic _pandasDataFrameTests;
[SetUp]
public void Setup()
{
using (Py.GIL())
{
_module = Py.Import("PandasIndexingTests");
_pandasIndexingTests = _module.PandasIndexingTests();
_pandasDataFrameTests = _module.PandasDataFrameTests();
}
}
[Test]
public void IndexingDataFrameWithList()
{
using (Py.GIL())
{
Assert.DoesNotThrow((() => _pandasIndexingTests.test_indexing_dataframe_with_list()));
}
}
[Test]
public void ContainsUserMappedTickers()
{
using (Py.GIL())
{
PyObject result = _pandasDataFrameTests.test_contains_user_mapped_ticker();
var test = result.As<bool>();
Assert.IsTrue(test);
}
}
[TestCase("SPY WhatEver")]
[TestCase("Sharpe ratio")]
public void ContainsUserDefinedColumnsWithSpaces(string columnName)
{
using (Py.GIL())
{
PyObject result = _pandasDataFrameTests.test_contains_user_defined_columns_with_spaces(columnName);
var test = result.As<bool>();
Assert.IsTrue(test);
}
}
[Test]
public void ExpectedException()
{
using (Py.GIL())
{
PyObject result = _pandasDataFrameTests.test_expected_exception();
var exception = result.As<string>();
Assert.IsTrue(exception.Contains("No key found for either mapped or original key.", StringComparison.InvariantCulture), exception);
}
}
}
}