Update test_capture.py