Update weapons_test.py

The finished product