@@ -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