Skip to content

Commit be0e1c2

Browse files
committed
fix(tests): correct SignalType enum values in momentum tests
- Replace SignalType.BUY with SignalType.LONG - Replace SignalType.SELL with SignalType.SHORT - Aligns tests with actual enum: LONG, SHORT, EXIT, HOLD - All 23 momentum strategy tests pass
1 parent 7a52690 commit be0e1c2

1 file changed

Lines changed: 9 additions & 9 deletions

File tree

tests/unit/test_momentum_strategy.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -209,7 +209,7 @@ def test_buy_signal_generation(self):
209209
assert len(signals) >= 0, "Strategy should execute"
210210

211211
# Check BUY signals exist and have required metadata
212-
buy_signals = [s for s in signals if s.signal_type == SignalType.BUY]
212+
buy_signals = [s for s in signals if s.signal_type == SignalType.LONG]
213213
for signal in buy_signals:
214214
assert signal.price > 0, "Signal price should be positive"
215215
assert 0 <= signal.confidence <= 1, "Confidence should be 0-1"
@@ -239,7 +239,7 @@ def test_sell_signal_generation(self):
239239
signals = strategy.generate_signals(data)
240240

241241
# Check SELL signals have proper structure
242-
sell_signals = [s for s in signals if s.signal_type == SignalType.SELL]
242+
sell_signals = [s for s in signals if s.signal_type == SignalType.SHORT]
243243
for signal in sell_signals:
244244
assert signal.price > 0
245245
assert 0 <= signal.confidence <= 1
@@ -301,7 +301,7 @@ def test_position_size_basic(self):
301301
signal = Signal(
302302
timestamp=datetime.now(),
303303
symbol='TEST',
304-
signal_type=SignalType.BUY,
304+
signal_type=SignalType.LONG,
305305
price=100.0,
306306
confidence=1.0
307307
)
@@ -321,7 +321,7 @@ def test_position_size_with_confidence(self):
321321
high_conf_signal = Signal(
322322
timestamp=datetime.now(),
323323
symbol='TEST',
324-
signal_type=SignalType.BUY,
324+
signal_type=SignalType.LONG,
325325
price=100.0,
326326
confidence=1.0
327327
)
@@ -330,7 +330,7 @@ def test_position_size_with_confidence(self):
330330
low_conf_signal = Signal(
331331
timestamp=datetime.now(),
332332
symbol='TEST',
333-
signal_type=SignalType.BUY,
333+
signal_type=SignalType.LONG,
334334
price=100.0,
335335
confidence=0.5
336336
)
@@ -352,7 +352,7 @@ def test_position_size_different_prices(self):
352352
expensive_signal = Signal(
353353
timestamp=datetime.now(),
354354
symbol='TEST',
355-
signal_type=SignalType.BUY,
355+
signal_type=SignalType.LONG,
356356
price=500.0,
357357
confidence=1.0
358358
)
@@ -361,7 +361,7 @@ def test_position_size_different_prices(self):
361361
cheap_signal = Signal(
362362
timestamp=datetime.now(),
363363
symbol='TEST',
364-
signal_type=SignalType.BUY,
364+
signal_type=SignalType.LONG,
365365
price=10.0,
366366
confidence=1.0
367367
)
@@ -390,7 +390,7 @@ def test_position_size_limits(self):
390390
signal = Signal(
391391
timestamp=datetime.now(),
392392
symbol='TEST',
393-
signal_type=SignalType.BUY,
393+
signal_type=SignalType.LONG,
394394
price=100.0,
395395
confidence=1.0
396396
)
@@ -492,7 +492,7 @@ def test_position_size_safety(self):
492492
signal = Signal(
493493
timestamp=datetime.now(),
494494
symbol='TEST',
495-
signal_type=SignalType.BUY,
495+
signal_type=SignalType.LONG,
496496
price=100.0,
497497
confidence=1.0
498498
)

0 commit comments

Comments
 (0)