[Código] Teorema de Pítagoras

########################
# Teorema de Pitagoras #
# Creador por: S&S Lab #
# http://artehack.net/ #
########################

import math

def Hipotenusa(X,Y):
    return math.sqrt((X*X)+(Y*Y))

def Cateto(X,Y):
    return math.sqrt((X*X)-(Y*Y))

 

print """
    ==== Teorema de Pitagoras ====
    1 - Hallar valor de un cateto
    2 - Hallar valor de hipotenusa
    ==============================
    """
opcion = input('Elija la opcion deseada > ')
if opcion == 1:
    X = input('Introduce el valor del primer cateto > ')
    Y = input('Introduce el valor del segundo cateto > ')
    print 'El valor de la hipotenusa es > %s' % Hipotenusa(X,Y)
if opcion == 2:
    X = input('Introduce el valor de un cateto > ')
    Y = input('Introduce el valor de la hipotenusa > ')
    print 'El valor del cateto es > %s' % Cateto(X,Y)
if (opcion <= 0) or (opcion >= 3):
    exit()

print """
       ==== Teorema de Pitagoras ====
       1 - Hallar valor de hipotenusa
       2 - Hallar valor de un cateto
       ==============================
       """
opcion = input('Elija la opcion deseada > ')