refector test to use assertEqual