Skip to content

Commit 5fba19b

Browse files
committed
Fix wol_pids test to compare proper pids.
1 parent 1f8fb21 commit 5fba19b

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

test_replays/test_all.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -191,10 +191,7 @@ def test_hots_pids():
191191
def test_wol_pids():
192192
replay = sc2reader.load_replay("test_replays/1.5.4.24540/ggtracker_1471849.SC2Replay")
193193

194-
efilter = lambda e: e.name == "AbilityEvent"
195-
abilityevents = filter(efilter, replay.events)
196-
ability_pids = set( [ event.pid for event in abilityevents ] )
197-
194+
ability_pids = set( [ event.player.pid for event in replay.events if 'AbilityEvent' in event.name ] )
198195
player_pids = set( [ player.pid for player in replay.players ] )
199196

200197
assert ability_pids == player_pids

0 commit comments

Comments
 (0)