Sollicitatievraag bij Tesla

Write some test cases based on the given code