Added tests for battle_engine.py