test: Add unit tests for new indicators including DSP, LRSI, PO, TRIXH, VWMACD, MMAR, RAINBOW, drawdown, HWC, and VFI

Author: xgboosted

tests/test_ext_indicator_cycles.py

@@ -21,6 +21,11 @@ def setUp(self):
     def tearDown(self):
         pass
 
+    def test_dsp_ext(self):
+        self.data.ta.dsp(append=True)
+        self.assertIsInstance(self.data, DataFrame)
+        self.assertEqual(self.data.columns[-1], "DSP_14")
+
     def test_ebsw_ext(self):
         self.data.ta.ebsw(append=True)
         self.assertIsInstance(self.data, DataFrame)

tests/test_ext_indicator_momentum.py

@@ -304,3 +304,28 @@ def test_willr_ext(self):
         self.data.ta.willr(append=True)
         self.assertIsInstance(self.data, DataFrame)
         self.assertEqual(self.data.columns[-1], "WILLR_14")
+
+    def test_lrsi_ext(self):
+        self.data.ta.lrsi(append=True)
+        self.assertIsInstance(self.data, DataFrame)
+        self.assertEqual(self.data.columns[-1], "LRSI_14")
+
+    def test_po_ext(self):
+        self.data.ta.po(append=True)
+        self.assertIsInstance(self.data, DataFrame)
+        self.assertEqual(self.data.columns[-1], "PO_14")
+
+    def test_trixh_ext(self):
+        self.data.ta.trixh(append=True)
+        self.assertIsInstance(self.data, DataFrame)
+        self.assertEqual(
+            list(self.data.columns[-3:]), ["TRIX_18_9", "TRIXs_18_9", "TRIXh_18_9"]
+        )
+
+    def test_vwmacd_ext(self):
+        self.data.ta.vwmacd(append=True)
+        self.assertIsInstance(self.data, DataFrame)
+        self.assertEqual(
+            list(self.data.columns[-3:]),
+            ["VWMACD_12_26_9", "VWMACDh_12_26_9", "VWMACDs_12_26_9"],
+        )

tests/test_ext_indicator_overlap_ext.py

@@ -196,3 +196,17 @@ def test_zlma_ext(self):
         self.data.ta.zlma(append=True)
         self.assertIsInstance(self.data, DataFrame)
         self.assertEqual(self.data.columns[-1], "ZL_EMA_10")
+
+    def test_mmar_ext(self):
+        self.data.ta.mmar(append=True)
+        self.assertIsInstance(self.data, DataFrame)
+        # MMAR returns multiple columns, check if any MMAR columns were added
+        mmar_cols = [col for col in self.data.columns if col.startswith("MMAR_")]
+        self.assertTrue(len(mmar_cols) > 0)
+
+    def test_rainbow_ext(self):
+        self.data.ta.rainbow(append=True)
+        self.assertIsInstance(self.data, DataFrame)
+        # RAINBOW returns multiple columns, check if any RAINBOW columns were added
+        rainbow_cols = [col for col in self.data.columns if col.startswith("RAINBOW_")]
+        self.assertTrue(len(rainbow_cols) > 0)

tests/test_ext_indicator_performance.py

@@ -41,3 +41,9 @@ def test_cum_percent_return_ext(self):
         self.data.ta.percent_return(append=True, cumulative=True)
         self.assertIsInstance(self.data, DataFrame)
         self.assertEqual(self.data.columns[-1], "CUMPCTRET_1")
+
+    def test_drawdown_ext(self):
+        self.data.ta.drawdown(append=True)
+        self.assertIsInstance(self.data, DataFrame)
+        # Drawdown returns 3 columns: DD, DD_PCT, DD_LOG
+        self.assertEqual(list(self.data.columns[-3:]), ["DD", "DD_PCT", "DD_LOG"])

tests/test_ext_indicator_trend.py

@@ -134,3 +134,23 @@ def test_vortext_ext(self):
         self.data.ta.vortex(append=True)
         self.assertIsInstance(self.data, DataFrame)
         self.assertEqual(list(self.data.columns[-2:]), ["VTXP_14", "VTXM_14"])
+
+    def test_pmax_ext(self):
+        self.data.ta.pmax(append=True)
+        self.assertIsInstance(self.data, DataFrame)
+        self.assertEqual(self.data.columns[-1], "PMAX_E_10_3.0")
+
+    def test_tsignals_ext(self):
+        # Create a simple trend series for testing
+        trend = self.data.ta.sma(length=10) - self.data.ta.sma(length=20)
+        trend = (trend > 0).astype(int)
+        result = self.data.ta.tsignals(trend)
+        self.assertIsInstance(result, DataFrame)
+        self.assertEqual(result.name, "TS")
+
+    def test_xsignals_ext(self):
+        # Create simple signal series for testing
+        signal = self.data.ta.rsi()
+        result = self.data.ta.xsignals(signal, xa=70, xb=30)
+        self.assertIsInstance(result, DataFrame)
+        self.assertEqual(result.name, "XS")

tests/test_ext_indicator_volatility.py

@@ -118,3 +118,10 @@ def test_ui_ext(self):
         self.data.ta.ui(append=True, everget=True)
         self.assertIsInstance(self.data, DataFrame)
         self.assertEqual(self.data.columns[-1], "UIe_14")
+
+    def test_hwc_ext(self):
+        self.data.ta.hwc(append=True)
+        self.assertIsInstance(self.data, DataFrame)
+        # HWC returns 3 columns: HWM, HWU, HWL
+        hwc_cols = [col for col in self.data.columns if col.startswith("HW")]
+        self.assertTrue(len(hwc_cols) >= 3)

tests/test_ext_indicator_volume.py

@@ -116,3 +116,8 @@ def test_vp_ext(self):
         result = self.data.ta.vp()
         self.assertIsInstance(result, DataFrame)
         self.assertEqual(result.name, "VP_10")
+
+    def test_vfi_ext(self):
+        self.data.ta.vfi(append=True)
+        self.assertIsInstance(self.data, DataFrame)
+        self.assertEqual(self.data.columns[-1], "VFI_130")

tests/test_indicator_cycles.py

@@ -48,6 +48,11 @@ def setUp(self):
     def tearDown(self):
         pass
 
+    def test_dsp(self):
+        result = pandas_ta.dsp(self.close)
+        self.assertIsInstance(result, Series)
+        self.assertEqual(result.name, "DSP_14")
+
     def test_ebsw(self):
         result = pandas_ta.ebsw(self.close)
         self.assertIsInstance(result, Series)

tests/test_indicator_momentum.py

@@ -600,6 +600,26 @@ def test_willr(self):
             except Exception as ex:
                 error_analysis(result, CORRELATION, ex)
 
+    def test_lrsi(self):
+        result = pandas_ta.lrsi(self.close)
+        self.assertIsInstance(result, Series)
+        self.assertEqual(result.name, "LRSI_14")
+
+    def test_po(self):
+        result = pandas_ta.po(self.close)
+        self.assertIsInstance(result, Series)
+        self.assertEqual(result.name, "PO_14")
+
+    def test_trixh(self):
+        result = pandas_ta.trixh(self.close)
+        self.assertIsInstance(result, DataFrame)
+        self.assertEqual(result.name, "TRIXH_18_9")
+
+    def test_vwmacd(self):
+        result = pandas_ta.vwmacd(self.close, self.volume)
+        self.assertIsInstance(result, DataFrame)
+        self.assertEqual(result.name, "VWMACD_12_26_9")
+
         result = pandas_ta.willr(self.high, self.low, self.close)
         self.assertIsInstance(result, Series)
         self.assertEqual(result.name, "WILLR_14")

tests/test_indicator_overlap.py

@@ -508,3 +508,13 @@ def test_zlma(self):
         result = pandas_ta.zlma(self.close)
         self.assertIsInstance(result, Series)
         self.assertEqual(result.name, "ZL_EMA_10")
+
+    def test_mmar(self):
+        result = pandas_ta.mmar(self.close)
+        self.assertIsInstance(result, DataFrame)
+        self.assertEqual(result.name, "MMAR_10_5_6")
+
+    def test_rainbow(self):
+        result = pandas_ta.rainbow(self.close)
+        self.assertIsInstance(result, DataFrame)
+        self.assertEqual(result.name, "RAINBOW_2_10")