In $[85]$: import math

import random

import numpy as $np$

from scipy.stats import binom

from scipy.stats import norm

from random import shuffle

success$\_$count $=0$

num$\_$simulations $=10000$

for $i$ in range$($num$\_$simulations$)$:

boxes $=$ random. sample$(range(1,11),10)$

found$\_$count $=0$

for $j$ in range $(10)$:

current$\_$box $=$ boxes $j$

if current$\_$box $==j+1$ :

found$\_$count $+=1$

else:

for $k$ in range $(4)$:

current$\_$box $=$ boxes $[$ current$\_$box$-1]$

if current$\_$box $==j+1$ :

found count $+=1$

break

if found$\_$count $==10$ :

success$\_$count $+=1$

break

prob$\_$of$\_$success $=$ success$\_$count $/$ num$\_$simulations

print $($prob of success$)$

Code gives assertion error, what can be fixed? Thank you