Cascade Data Labs interview question

Take home assignment with test case