diff --git a/sc2reader/engine/plugins/creeptracker.py b/sc2reader/engine/plugins/creeptracker.py index 788f5806..7f35f868 100644 --- a/sc2reader/engine/plugins/creeptracker.py +++ b/sc2reader/engine/plugins/creeptracker.py @@ -77,11 +77,12 @@ def handleEndGame(self, event, replay): if player.play_race[0] == 'Z': self.creepTracker.reduce_cgu_per_minute(player.pid) player.creep_spread_by_minute = self.creepTracker.get_creep_spread_area(player.pid) + # note that player.max_creep_spread may be a tuple or an int if player.creep_spread_by_minute: player.max_creep_spread = max(player.creep_spread_by_minute.items(),key=lambda x:x[1]) else: ## Else statement is for players with no creep spread(ie: not Zerg) - player.max_creep_spread =0 + player.max_creep_spread = 0 except Exception as e: print("Whoa! {}".format(e)) pass diff --git a/test_replays/test_all.py b/test_replays/test_all.py index 0cc12077..d04e2089 100644 --- a/test_replays/test_all.py +++ b/test_replays/test_all.py @@ -431,7 +431,7 @@ def test_lotv_creepTracker(self): for player_id in replay.player: if replay.player[player_id].play_race == "Zerg": - assert replay.player[player_id].max_creep_spread >0 + assert replay.player[player_id].max_creep_spread != 0 assert replay.player[player_id].creep_spread_by_minute def test_lotv_map(self):