Update media_tests.py