Skip to content

Commit

Permalink
fix:tests for peek_by_condition
Browse files Browse the repository at this point in the history
  • Loading branch information
galenseilis committed Dec 8, 2024
1 parent 2a3bb81 commit c6ee366
Showing 1 changed file with 18 additions and 17 deletions.
35 changes: 18 additions & 17 deletions tests/test_eventscheduler_class/test_peek_by_condition.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,24 +14,24 @@ def test_next_by_time():

# NOTE: Cases related to t=-1
assert env.peek_by_condition(lambda env, event: event.time < -1) is None
assert env.next_event_by_condition(lambda env, event: event.time <= -1) == event0
assert env.next_event_by_condition(lambda env, event: event.time == -1) == event0
assert env.next_event_by_condition(lambda env, event: event.time >= -1) == event0
assert env.next_event_by_condition(lambda env, event: event.time > -1) is event1
assert env.peek_by_condition(lambda env, event: event.time <= -1) == -1
assert env.peek_by_condition(lambda env, event: event.time == -1) == -1
assert env.peek_by_condition(lambda env, event: event.time >= -1) == -1
assert env.peek_by_condition(lambda env, event: event.time > -1) is 0

# NOTE: Cases related to t=0
assert env.next_event_by_condition(lambda env, event: event.time < 0) is event0
assert env.next_event_by_condition(lambda env, event: event.time <= 0) == event0
assert env.next_event_by_condition(lambda env, event: event.time == 0) == event1
assert env.next_event_by_condition(lambda env, event: event.time >= 0) == event1
assert env.next_event_by_condition(lambda env, event: event.time > 0) is event2
assert env.peek_by_condition(lambda env, event: event.time < 0) is -1
assert env.peek_by_condition(lambda env, event: event.time <= 0) == -1
assert env.peek_by_condition(lambda env, event: event.time == 0) == 0
assert env.peek_by_condition(lambda env, event: event.time >= 0) == 0
assert env.peek_by_condition(lambda env, event: event.time > 0) is 1

# NOTE: Cases related to t=1
assert env.next_event_by_condition(lambda env, event: event.time < 1) is event0
assert env.next_event_by_condition(lambda env, event: event.time <= 1) == event0
assert env.next_event_by_condition(lambda env, event: event.time == 1) == event2
assert env.next_event_by_condition(lambda env, event: event.time >= 1) == event2
assert env.next_event_by_condition(lambda env, event: event.time > 1) is None
assert env.peek_by_condition(lambda env, event: event.time < 1) is -1
assert env.peek_by_condition(lambda env, event: event.time <= 1) == -1
assert env.peek_by_condition(lambda env, event: event.time == 1) == 1
assert env.peek_by_condition(lambda env, event: event.time >= 1) == 1
assert env.peek_by_condition(lambda env, event: event.time > 1) is None


def test_next_by_if_action():
Expand All @@ -49,19 +49,20 @@ def test_next_by_if_action():
assert condition0(env, event0) == True
assert condition0(env, event1) == True
assert condition0(env, event2) == False
assert env.next_event_by_condition(condition0) == event0
assert env.peek_by_condition(condition0) == 0

condition1 = lambda env, event: event.action() == 2018
assert condition1(env, event0) == False
assert condition1(env, event1) == False
assert condition1(env, event2) == True
assert env.next_event_by_condition(condition1) == event2
assert env.peek_by_condition(condition1) == 0

condition2 = lambda env, event: event.action == 2018
assert condition2(env, event0) == False
assert condition2(env, event1) == False
assert condition2(env, event2) == False
assert env.next_event_by_condition(condition2) is None
assert env.peek_by_condition(condition2) is None


# TODO: Add more test cases...

0 comments on commit c6ee366

Please sign in to comment.