remove print in test
Update comment and add doc about the new warning in the infer_shape tester.