помогите пожалуйста написать unittest для это программы, ну очень нужно


Код Python

def IsPrime(n):
for i in range(2,int(n ** 0.5)+1):
if n % i == 0:
return False
return True
a=int(input())
res=
i=2
while IsPrime(a)==0:
if IsPrime(i)==1 and a%i==0:
a=int(a/i)
res.append(i)
i=2
i+=1
res.append(a)
print(sorted(res))

Есть ф-ия, которая дает список множителей из числа, то достаточно перемножить их все между собой, и сравнить с оригинальным числом. Если произведение и начальное число совпадают - тест пройден.

Понимаю, что это просто, но с программированием у меня ваще беда((