Create test_with_class.py