Python : check prime

sir what is the mistake in my code

it is passing only one test case

Please check your code it is not to check prime

Note :

also ask doubt on the question which you want to ask
so that we can also see your submission(code)

you have asked doubt on “simple-input” but actually your doubt is on check prime question