test case http://stackoverflow.com/q/30415074/128629

test.py

import random
def f():
    total_damage = 0 
    attack_score = 0
    defence_score = 0
    enemy_list = list(range(10))
    power = 5
    unit_size = 2000
    enemy_defence = 5
    enemy_unit_size = 2000
    """
    for team in teams:
        for unit in team:
            if unit.team != self.team:
                enemy_list.append(unit)
    """
    for enemy in enemy_list:
        if True:
            for i in range(int(power*unit_size)):
                attack_score += random.randint(0, power)
            for i in range(int(enemy_defence*enemy_unit_size)):
                defence_score += random.randint(0, enemy_defence)
            if attack_score > defence_score:
                total_damage = attack_score - defence_score
            else:
                total_damage = 0


import time
before = time.time()
f()
after = time.time()
print(after-before)