Cálculo de la hipotenusa o el cateto

Hoy, al tener tarde libre tenía un poco de tiempo libre y con la ayuda de David he creado el programa que ya quería hacer antes.
El programa calculaba la hipotenusa de un triangulo rectangulo del que le das los dos catetos o un cateto si les das la hipotenusa y el otro cateto.
Al fin he descubierto que me saltaba un r al teclear el código de la raiz cuadrada y por eso no funcionaba y me he dedicado a jugar con el if y el else.
El código que me ha quedado es el siguiente:

lado=float(raw_input("Lado que falta (para hipotenusa pulse 1, para cateto 2): "))
if lado==1:
a=float(raw_input("cateto1: "))
b=float(raw_input("cateto2: "))
c=(raw_input("unidad de medida: "))
import math
r=a**2+b**2
print "La hipotenusa mide", math.sqrt(r),c
else:
d=float(raw_input("hipotenusa: "))
e=float(raw_input("cateto: "))
f=(raw_input("unidad de medida: "))
import math
r=d**2-e**2
print "La hipotenusa mide", math.sqrt(r),f

(Hasta te pone la unidad de medida xD)
La verdad es que me parece bastante más útil que el otro, no mucho porque con la calculadora se hace más rápido.