change to use assertEqual