Corrigido erro do EntityUnit e criado o test.py
Renomeado test.py
Retirada uma inclusão para deixar o código mais reusável.