Один математический совет #Python :

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

Вместо этого используйте 𝚖𝚊𝚝𝚑.𝚒𝚜𝚌𝚕𝚘𝚜𝚎().