Update enemy_test.py