W treści zadania jest podane ile testów n ma być wykonane czyli po podaniu liczby n (przez użytkownika), program powinien "pobrać" od użytkownika n liczb, czyli ten zapis z def czy_pierwsza(A):
if A <= 1 or A >= 100000:
return "NIE"
powinien zniknąć, ponad to (dla przypomnienia )
Liczba pierwsza - liczba naturalna większa od 1, ...
zmienił bym zapis
if A == 2:
return "TAK"
na
if A <= 1:
return "NIE"
wtedy mamy
def czy_pierwsza(A):
if A <= 1:
return "NIE"
for x in range(2, A):
if A % x == 0:
return "NIE"
return "TAK"
Przykład dla pobrania n testów
print("Podaj liczbę testów [1...100000]")
n = int(input())
if n < 1 or n > 100000:
print("Podana liczba testów jest nieprawidłowa")
else:
A = []
for i in range(n):
A.append(int(input()))
for i in A:
print(czy_pierwsza(i))