from deltatrace.event_graph import Event, EventType, EventGraph
from deltatrace.replay_engine import ReplayEngine


def test_replay_engine_basic():
    g = EventGraph()
    t1 = Event.create(EventType.MD_TICK, {"v": 1})
    s1 = Event.create(EventType.SIGNAL, {"v": 2})
    g.add_event(t1)
    g.add_event(s1)
    g.add_edge(t1.id, s1.id)

    engine = ReplayEngine()
    res = engine.replay(g)
    assert res.events_replayed == 2
    assert 0.0 <= res.fidelity_score <= 1.0