primetest.py

def return_prime(num):

    prime = "is prime."
    not_prime = "is not prime."
    divisors = 0

    if num > 1:
        for i in range(2,num):
            if (num % i) == 0:
                divisors = divisors + 1
            else:
                pass
        if divisors > 0:
            return not_prime
        else:
            return prime
    else:
        return not_prime

def main():

    five = return_prime(5)
    four = return_prime(4)

    print("Five ", five)
    print("Four ", four)

if __name__ == '__main__':
    main()