Add unit tests by using assert