Como calcular o Momento de Inércia de um quadrado em Python - Fórmula do Momento de Inércia de um quadrado

# vamos importar o módulo Math
import math

# função principal do programa
def main():
  # vamos pedir para o usuário informar o valor dos lados do quadrado
  h = float(input("Valor dos lados do quadrado em centímetros: "))

  # vamos calcular o momento de inércia em relação ao eixo x
  momento_inercia_x = math.pow(h, 4) / 12
  # vamos converter para centímetros à quarta
  momento_inercia_x = momento_inercia_x / 1000

  # vamos calcular o momento de inércia em relação ao eixo y
  momento_inercia_y = math.pow(h, 4) / 12
  # vamos converter para centímetros à quarta
  momento_inercia_y = momento_inercia_y / 1000

  # vamos mostrar os resultados
  print("O momento de inércia em relação ao eixo x é: {0} cm4".format(
    momento_inercia_x))
  print("O momento de inércia em relação ao eixo y é: {0} cm4".format(
    momento_inercia_y))

if __name__== "__main__":
  main()