diff options
Diffstat (limited to 'shared/tests/test_events.py')
| -rw-r--r-- | shared/tests/test_events.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/shared/tests/test_events.py b/shared/tests/test_events.py index 4bc7981..ab7792b 100644 --- a/shared/tests/test_events.py +++ b/shared/tests/test_events.py @@ -1,11 +1,12 @@ """Tests for shared event types.""" -import pytest + from decimal import Decimal from datetime import datetime, timezone def make_candle(): from shared.models import Candle + return Candle( symbol="BTCUSDT", timeframe="1m", @@ -20,6 +21,7 @@ def make_candle(): def make_signal(): from shared.models import Signal, OrderSide + return Signal( strategy="test", symbol="BTCUSDT", @@ -33,6 +35,7 @@ def make_signal(): def test_candle_event_serialize(): """Test CandleEvent serializes to dict correctly.""" from shared.events import CandleEvent, EventType + candle = make_candle() event = CandleEvent(data=candle) d = event.to_dict() @@ -44,6 +47,7 @@ def test_candle_event_serialize(): def test_candle_event_deserialize(): """Test CandleEvent round-trips through to_dict/from_raw.""" from shared.events import CandleEvent, EventType + candle = make_candle() event = CandleEvent(data=candle) d = event.to_dict() @@ -56,6 +60,7 @@ def test_candle_event_deserialize(): def test_signal_event_serialize(): """Test SignalEvent serializes to dict correctly.""" from shared.events import SignalEvent, EventType + signal = make_signal() event = SignalEvent(data=signal) d = event.to_dict() @@ -66,7 +71,8 @@ def test_signal_event_serialize(): def test_event_from_dict_dispatch(): """Test Event.from_dict dispatches to correct class.""" - from shared.events import Event, CandleEvent, SignalEvent, EventType + from shared.events import Event, CandleEvent, SignalEvent + candle = make_candle() event = CandleEvent(data=candle) d = event.to_dict() |
