Fix remaining unit test entries

twopirllc · Oct 13, 2023 · d60fc1a · d60fc1a
1 parent 9838e78
commit d60fc1a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/pandas_ta/momentum/tmo.py b/pandas_ta/momentum/tmo.py
@@ -134,7 +134,7 @@ def tmo(
 
     main_signal.name = f"TMO_{params}"
     smooth_signal.name = f"TMO_Smooth_{params}"
-    mom_main.name = f"TMO_Main_Mom_{params}"
+    mom_main.name = f"TMO_Mom_{params}"
     mom_smooth.name = f"TMO_Smooth_Mom_{params}"
 
     df = DataFrame({
@@ -146,4 +146,4 @@ def tmo(
     df.name = f"TMO_{params}"
     df.category = tmo_category
 
-    return df
+    return df
diff --git a/tests/test_indicator_momentum.py b/tests/test_indicator_momentum.py
@@ -778,10 +778,10 @@ def test_ext_td_seq(df):
     assert list(df.columns[-2:]) == ["TD_SEQ_UPa", "TD_SEQ_DNa"]
 
 
-def test_tmo(self):
-    result = pandas_ta.tmo(self.open, self.close)
-    self.assertIsInstance(result, DataFrame)
-    self.assertEqual(result.name, "TMO_14_5_3")
+def test_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"]
+    assert list(df.columns[-4:]) == "columns"
 
 
 def test_ext_trix(df):