Update snorkel_tests.py