diff options
Diffstat (limited to 'tests/db_functions/math/test_floor.py')
| -rw-r--r-- | tests/db_functions/math/test_floor.py | 25 |
1 files changed, 14 insertions, 11 deletions
diff --git a/tests/db_functions/math/test_floor.py b/tests/db_functions/math/test_floor.py index 1068e55476..b660c1c48e 100644 --- a/tests/db_functions/math/test_floor.py +++ b/tests/db_functions/math/test_floor.py @@ -10,15 +10,16 @@ from ..models import DecimalModel, FloatModel, IntegerModel class FloorTests(TestCase): - def test_null(self): IntegerModel.objects.create() - obj = IntegerModel.objects.annotate(null_floor=Floor('normal')).first() + obj = IntegerModel.objects.annotate(null_floor=Floor("normal")).first() self.assertIsNone(obj.null_floor) def test_decimal(self): - DecimalModel.objects.create(n1=Decimal('-12.9'), n2=Decimal('0.6')) - obj = DecimalModel.objects.annotate(n1_floor=Floor('n1'), n2_floor=Floor('n2')).first() + DecimalModel.objects.create(n1=Decimal("-12.9"), n2=Decimal("0.6")) + obj = DecimalModel.objects.annotate( + n1_floor=Floor("n1"), n2_floor=Floor("n2") + ).first() self.assertIsInstance(obj.n1_floor, Decimal) self.assertIsInstance(obj.n2_floor, Decimal) self.assertEqual(obj.n1_floor, Decimal(math.floor(obj.n1))) @@ -26,7 +27,9 @@ class FloorTests(TestCase): def test_float(self): FloatModel.objects.create(f1=-27.5, f2=0.33) - obj = FloatModel.objects.annotate(f1_floor=Floor('f1'), f2_floor=Floor('f2')).first() + obj = FloatModel.objects.annotate( + f1_floor=Floor("f1"), f2_floor=Floor("f2") + ).first() self.assertIsInstance(obj.f1_floor, float) self.assertIsInstance(obj.f2_floor, float) self.assertEqual(obj.f1_floor, math.floor(obj.f1)) @@ -35,9 +38,9 @@ class FloorTests(TestCase): def test_integer(self): IntegerModel.objects.create(small=-20, normal=15, big=-1) obj = IntegerModel.objects.annotate( - small_floor=Floor('small'), - normal_floor=Floor('normal'), - big_floor=Floor('big'), + small_floor=Floor("small"), + normal_floor=Floor("normal"), + big_floor=Floor("big"), ).first() self.assertIsInstance(obj.small_floor, int) self.assertIsInstance(obj.normal_floor, int) @@ -48,7 +51,7 @@ class FloorTests(TestCase): def test_transform(self): with register_lookup(DecimalField, Floor): - DecimalModel.objects.create(n1=Decimal('5.4'), n2=Decimal('0')) - DecimalModel.objects.create(n1=Decimal('3.4'), n2=Decimal('0')) + DecimalModel.objects.create(n1=Decimal("5.4"), n2=Decimal("0")) + DecimalModel.objects.create(n1=Decimal("3.4"), n2=Decimal("0")) obj = DecimalModel.objects.filter(n1__floor__gt=4).get() - self.assertEqual(obj.n1, Decimal('5.4')) + self.assertEqual(obj.n1, Decimal("5.4")) |