Mi código de pruebas en python