Replace deprecated assertEquals with assertEqual