Update test_with_class.py