chore: fix laguerre - it should return a series like all others

freqtrade · Aug 3, 2024 · 7d4915c · 7d4915c
1 parent 4b17cb5
commit 7d4915c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/technical/indicators/indicators.py b/technical/indicators/indicators.py
@@ -95,7 +95,7 @@ def ichimoku(
 #
 # Laguerre RSI
 #
-def laguerre(dataframe, gamma=0.75, smooth=1, debug=bool):
+def laguerre(dataframe, gamma=0.75, smooth=1, debug=bool) -> Series:
     """
     laguerre RSI
     Author Creslin
@@ -179,7 +179,7 @@ def laguerre(dataframe, gamma=0.75, smooth=1, debug=bool):
         else:
             lrsi_l.append(0)
 
-    return lrsi_l
+    return Series(lrsi_l)
 
 
 ########################################

diff --git a/tests/test_indicators_generic.py b/tests/test_indicators_generic.py
@@ -25,7 +25,7 @@
         (ti.fibonacci_retracements, [], "series", None),
         (ti.hull_moving_average, [10], "series", None),
         (ti.ichimoku, [], "dict", None),
-        (ti.laguerre, [], "list", None),
+        (ti.laguerre, [], "series", None),
         (ti.madrid_sqz, [], "tuple", None),
         (ti.mmar, [], "tuple", None),
         (ti.osc, [], "series", None),
@@ -56,11 +56,6 @@ def test_indicators_generic_interface(
         assert isinstance(res, tuple)
         assert len(res[0]) == 1000
         assert len(res[1]) == 1000
-    elif responsetype == "list":
-        # only laguerre is this
-        # TODO: This should be changed!
-        assert isinstance(res, list)
-        assert len(res) == 1000
     elif responsetype == "dict":
         assert isinstance(res, dict)
         assert len(res["tenkan_sen"]) == 1000