From 41106d3c4961e67a2bc8a8d6d741ab1ca4541e11 Mon Sep 17 00:00:00 2001 From: Dmitry Novikov Date: Mon, 29 Jul 2019 02:44:36 +0300 Subject: [PATCH] simple refactoring --- tariff_app/models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tariff_app/models.py b/tariff_app/models.py index 44f7e0e..7ff47b9 100644 --- a/tariff_app/models.py +++ b/tariff_app/models.py @@ -88,14 +88,14 @@ class PeriodicPay(models.Model): # calc_obj = self._get_calc_object() res = calc_obj.get_next_time_to_pay(self, last_time_payment) - if type(res) is not datetime: + if not isinstance(res, datetime): raise TypeError return res def calc_amount(self): calc_obj = self._get_calc_object() res = calc_obj.calc_amount(self) - if type(res) is not float: + if not isinstance(res, float): raise TypeError return res