Fix extension unit test

twopirllc · Oct 13, 2023 · 6bfc5e0 · 6bfc5e0
1 parent d60fc1a
commit 6bfc5e0
Show file tree

Hide file tree

Showing 3 changed files with 81 additions and 81 deletions.
diff --git a/examples/TMO_indicator.ipynb b/examples/TMO_indicator.ipynb
diff --git a/pandas_ta/momentum/tmo.py b/pandas_ta/momentum/tmo.py
@@ -133,9 +133,9 @@ def tmo(
     params = f"{tmo_length}_{calc_length}_{smooth_length}"
 
     main_signal.name = f"TMO_{params}"
-    smooth_signal.name = f"TMO_Smooth_{params}"
-    mom_main.name = f"TMO_Mom_{params}"
-    mom_smooth.name = f"TMO_Smooth_Mom_{params}"
+    smooth_signal.name = f"TMOs_{params}"
+    mom_main.name = f"TMO_mom_{params}"
+    mom_smooth.name = f"TMOs_mom_{params}"
 
     df = DataFrame({
         main_signal.name: main_signal,

diff --git a/tests/test_indicator_momentum.py b/tests/test_indicator_momentum.py
@@ -778,9 +778,9 @@ def test_ext_td_seq(df):
     assert list(df.columns[-2:]) == ["TD_SEQ_UPa", "TD_SEQ_DNa"]
 
 
-def test_tmo(df):
+def test_ext_tmo(df):
     df.ta.tmo(append=True)
-    columns = ["TMO_14_5_3", "TMO_Smooth_14_5_3", "TMO_Mom_14_5_3", "TMO_Smooth_Mom_14_5_3"]
+    columns = ["TMO_14_5_3", "TMOs_14_5_3", "TMO_mom_14_5_3", "TMOs_mom_14_5_3"]
     assert list(df.columns[-4:]) == "columns"