Usando la función np.asarray al principio del cuerpo de la función se puede convertir la entrada a arrays de NumPy para que funcione código de este tipo:
>>> p, T, rho = isa.atm([0, 1000, 2000, 10000])
Hace falta añadir otro test más que valide esta funcionalidad.