Skip to content

Commit b2c51be

Browse files
committed
make test only care about pids matching
1 parent f4eb1fb commit b2c51be

File tree

1 file changed

+5
-9
lines changed

1 file changed

+5
-9
lines changed

test_replays/test_all.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -181,13 +181,9 @@ def test_datetimes():
181181

182182
def test_pids():
183183
replay = sc2reader.load_replay("test_replays/2.0.3.24764/Antiga Shipyard (3).SC2Replay")
184-
assert replay.players[0].pid == 0
185-
assert replay.player[0].pid == 0
186-
assert replay.players[1].pid == 6
187-
assert replay.player[6].pid == 6
188-
189-
eventpids = set()
190-
for event in replay.events:
191-
eventpids.add(event.pid)
184+
185+
player_pids = set( [ player.pid for player in replay.players ] )
186+
player_pids.add(16)
187+
event_pids = set( [ event.pid for event in replay.events ] )
192188

193-
assert eventpids == set([0, 6, 16])
189+
assert event_pids == player_pids

0 commit comments

Comments
 (0)