Update pressure_test.py

Added a line