Дано простое число. Создать функцию,которая находит следующее за ним простое число.
Есть код на паскале, но нужен на питоне:
var
n: integer;
function IsPrime(x: integer): boolean;
var
i: integer;
begin
IsPrime := true;
for i := 2 to Round(Sqrt(x)) do
if x mod i = 0 then
IsPrime := false;
end;
function NextPrime(x: integer): integer;
begin
if not IsPrime(x) then
exit
else
begin
repeat
Inc(x);
until IsPrime(x);
NextPrime := x;
end
end;
begin
repeat
Write('Введите простое число: ‘);
Readln(n);
until IsPrime(n);
Writeln(’Следующее простое = ', NextPrime(n));
Readln
end.