Make Python algorithms regression tests

QuantConnect · Jan 5, 2024 · 205f47a · 205f47a
1 parent 693cca5
commit 205f47a
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/Tests/Python/PythonWrapperTests.cs b/Tests/Python/PythonWrapperTests.cs
@@ -17,6 +17,7 @@
 using NUnit.Framework;
 using Python.Runtime;
 using QuantConnect.Python;
+using System.Collections.Generic;
 
 namespace QuantConnect.Tests.Python
 {
@@ -59,6 +60,26 @@ public void DoesNotThrowWhenDerivedFromCSharpModel()
                 }
             }
 
+            [Test]
+            public void SettlementModelPythonWrapperWorks()
+            {
+                var results = AlgorithmRunner.RunLocalBacktest("CustomSettlementModelRegressionAlgorithm",
+                new Dictionary<string, string>(),
+                Language.Python,
+                AlgorithmStatus.Completed,
+                algorithmLocation: "../../../Algorithm.Python/CustomSettlementModelRegressionAlgorithm.py");
+            }
+
+            [Test]
+            public void BenchmarkModelPythonWrapperWorks()
+            {
+                var results = AlgorithmRunner.RunLocalBacktest("CustomBenchmarkRegressionAlgorithm",
+                new Dictionary<string, string> { { "Total Trades", "1" } },
+                Language.Python,
+                AlgorithmStatus.Completed,
+                algorithmLocation: "../../../Algorithm.Python/CustomBenchmarkRegressionAlgorithm.py");
+            }
+
             private const string FullyImplemented =
                 @"
 from clr import AddReference