def test_hero_update_race(self):
"""tests setting race based on race of units"""
hero = Hero(user=self.user)
hero.save()
Unit(hero=hero,race='human').save()
Unit(hero=hero,race='elf').save()
Unit(hero=hero,race='elf').save()
hero.update_race()
assert hero.race == 'elf'
