Update class_tests.py