Assignment 1 Junit Test Solution