Update sample_tests.py