use assertEqual for test cases