Update wp_test.py