Помогите с заданием python
Написать функцию is_prime, принимающую 1 аргумент — число от 0 до 1000, и возвращающую и все переместить True, если оно простое, и False квадрат лишний и треугольник