TST: Change assertTrue to assertEquals