Atletico Madrid's Koke has said Vinicius Junior can expect "trouble" in Sunday's derby if he performs his trademark dancing celebration.