Unit tests for challenge5 of the python course