testing battle.py