Problem in pyhagoras triplet

#1

Out of the five test cases, my code is working correctly only with three test cases.
Can you please identify what is wrong with the code. LINK:

#2

I have edited your code, plz refer to it