Use assertEqual for assertions