Update wave_generator.py

Consertei essa bagaça, adiciona ai nerdão
Retornando booleano e inteiro?