Adding tests for battle_engine.py