fixed memory error in unit test and added a couple more distinct cases