Updating test_weights.py for naming convention