change all assert to assertEqual