Code Editor : test_assumed_shape.py