Code Signal Python Test Example