Переписана игра prime

Author: d4nich

brain_games/games/prime.py

@@ -1,12 +1,15 @@
 import random
 rules = 'Answer "yes" if given number is prime. Otherwise answer "no".'
+def is_prime(n):
+    if n < 2:
+        return False
+    for i in range(2, int(n ** 0.5) + 1):
+        if n % i == 0:
+            return False
+    return True
+
 def get_round():
     num = random.randint(1, 50)
     question = f'Question: {num}'
-    for i in range(1, num):
-        if num % i == 0 and i != num and i != 1:
-            correct_answer = 'no'
-            break
-        else:
-            correct_answer = 'yes'
+    correct_answer = 'yes' if is_prime(num) else 'no'
     return question, correct_answer

poetry.lock

@@ -1,6 +1,6 @@
 [project]
 name = "hexlet-code"
-version = "0.6.2"
+version = "0.6.3"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.10"