Fix issue with regression tests

QuantConnect · Dec 20, 2024 · 87bb775 · 87bb775
1 parent 7597d3c
commit 87bb775
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 6 deletions.
diff --git a/Indicators/IndicatorBase.cs b/Indicators/IndicatorBase.cs
@@ -253,10 +253,10 @@ protected IndicatorBase(string name)
         /// <returns>True if this indicator is ready, false otherwise</returns>
         public override bool Update(IBaseData input)
         {
-            if (typeof(T) == typeof(IndicatorDataPoint))
+            /* if (typeof(T) == typeof(IndicatorDataPoint))
             {
-                input = new IndicatorDataPoint(input.EndTime, input.Value);
-            }
+                input = (input.GetType() == typeof(IndicatorDataPoint)) ? input : new IndicatorDataPoint(input.EndTime, input.Value);
+            } */
             T _previousSymbolInput = default(T);
             if (_previousInput.TryGetValue(input.Symbol.ID, out _previousSymbolInput) && input.EndTime < _previousSymbolInput.EndTime)
             {
@@ -275,7 +275,14 @@ public override bool Update(IBaseData input)
 
                 if (!(input is T))
                 {
-                    throw new ArgumentException($"IndicatorBase.Update() 'input' expected to be of type {typeof(T)} but is of type {input.GetType()}");
+                    if (typeof(T) == typeof(IndicatorDataPoint))
+                    {
+                        input = new IndicatorDataPoint(input.EndTime, input.Value);
+                    }
+                    else
+                    {
+                        throw new ArgumentException($"IndicatorBase.Update() 'input' expected to be of type {typeof(T)} but is of type {input.GetType()}");
+                    }
                 }
                 _previousInput[input.Symbol.ID] = (T)input;
 

diff --git a/Tests/Indicators/SqueezeMomentumTests.cs b/Tests/Indicators/SqueezeMomentumTests.cs
@@ -27,7 +27,7 @@ protected override IndicatorBase<IBaseDataBar> CreateIndicator()
             VolumeRenkoBarSize = 0.5m;
             return new SqueezeMomentum("SM", 20, 2, 20, 1.5m);
         }
-        protected override string TestFileName => "spy_smi.csv";
+        protected override string TestFileName => "spy_sm.csv";
 
         protected override string TestColumnName => "squeeze on";
     }

diff --git a/Tests/QuantConnect.Tests.csproj b/Tests/QuantConnect.Tests.csproj
@@ -618,7 +618,7 @@
     <Content Include="TestData\spy_hurst_exponent.csv">
       <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
     </Content>
-    <Content Include="TestData\spy_smi.csv">
+    <Content Include="TestData\spy_sm.csv">
       <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
     </Content>
   </ItemGroup>

diff --git a/Tests/TestData/spy_smi.csv → Tests/TestData/spy_sm.csv b/Tests/TestData/spy_smi.csv → Tests/TestData/spy_sm.csv