fix: add missing choch_plus columns to TestGetChochBosStats test data

The test setUp was missing choch_plus_bullish and choch_plus_bearish
columns that get_choch_bos_stats() expects. Also updated assertions
to cover choch_plus counts and total_choch_plus key.

Pre-existing bug on main (not introduced by this PR).

Author: MDUYN

tests/indicators/test_market_structure.py

@@ -334,6 +334,8 @@ def setUp(self):
         self.df = pd.DataFrame({
             'choch_bullish': [0, 1, 0, 1, 0],
             'choch_bearish': [1, 0, 1, 0, 0],
+            'choch_plus_bullish': [0, 0, 0, 1, 0],
+            'choch_plus_bearish': [0, 1, 0, 0, 0],
             'bos_bullish': [0, 0, 1, 0, 1],
             'bos_bearish': [1, 0, 0, 1, 0]
         })
@@ -348,8 +350,10 @@ def test_get_choch_bos_stats_keys(self):
         result = get_choch_bos_stats(self.df)
         expected_keys = [
             'choch_bullish_count', 'choch_bearish_count',
+            'choch_plus_bullish_count',
+            'choch_plus_bearish_count',
             'bos_bullish_count', 'bos_bearish_count',
-            'total_choch', 'total_bos'
+            'total_choch', 'total_choch_plus', 'total_bos'
         ]
         for key in expected_keys:
             self.assertIn(key, result)
@@ -359,9 +363,12 @@ def test_get_choch_bos_stats_counts(self):
         result = get_choch_bos_stats(self.df)
         self.assertEqual(result['choch_bullish_count'], 2)
         self.assertEqual(result['choch_bearish_count'], 2)
+        self.assertEqual(result['choch_plus_bullish_count'], 1)
+        self.assertEqual(result['choch_plus_bearish_count'], 1)
         self.assertEqual(result['bos_bullish_count'], 2)
         self.assertEqual(result['bos_bearish_count'], 2)
         self.assertEqual(result['total_choch'], 4)
+        self.assertEqual(result['total_choch_plus'], 2)
         self.assertEqual(result['total_bos'], 4)