Update framework_test.py