Added infer_shape test for choose function.