Unit tests to solve the coding test