Use assertEqual