Create screenshot_test.py