diff options
Diffstat (limited to 'tests/db_functions/math/test_sign.py')
| -rw-r--r-- | tests/db_functions/math/test_sign.py | 29 |
1 files changed, 16 insertions, 13 deletions
diff --git a/tests/db_functions/math/test_sign.py b/tests/db_functions/math/test_sign.py index 0458932cee..c099441a19 100644 --- a/tests/db_functions/math/test_sign.py +++ b/tests/db_functions/math/test_sign.py @@ -9,23 +9,26 @@ from ..models import DecimalModel, FloatModel, IntegerModel class SignTests(TestCase): - def test_null(self): IntegerModel.objects.create() - obj = IntegerModel.objects.annotate(null_sign=Sign('normal')).first() + obj = IntegerModel.objects.annotate(null_sign=Sign("normal")).first() self.assertIsNone(obj.null_sign) def test_decimal(self): - DecimalModel.objects.create(n1=Decimal('-12.9'), n2=Decimal('0.6')) - obj = DecimalModel.objects.annotate(n1_sign=Sign('n1'), n2_sign=Sign('n2')).first() + DecimalModel.objects.create(n1=Decimal("-12.9"), n2=Decimal("0.6")) + obj = DecimalModel.objects.annotate( + n1_sign=Sign("n1"), n2_sign=Sign("n2") + ).first() self.assertIsInstance(obj.n1_sign, Decimal) self.assertIsInstance(obj.n2_sign, Decimal) - self.assertEqual(obj.n1_sign, Decimal('-1')) - self.assertEqual(obj.n2_sign, Decimal('1')) + self.assertEqual(obj.n1_sign, Decimal("-1")) + self.assertEqual(obj.n2_sign, Decimal("1")) def test_float(self): FloatModel.objects.create(f1=-27.5, f2=0.33) - obj = FloatModel.objects.annotate(f1_sign=Sign('f1'), f2_sign=Sign('f2')).first() + obj = FloatModel.objects.annotate( + f1_sign=Sign("f1"), f2_sign=Sign("f2") + ).first() self.assertIsInstance(obj.f1_sign, float) self.assertIsInstance(obj.f2_sign, float) self.assertEqual(obj.f1_sign, -1.0) @@ -34,9 +37,9 @@ class SignTests(TestCase): def test_integer(self): IntegerModel.objects.create(small=-20, normal=0, big=20) obj = IntegerModel.objects.annotate( - small_sign=Sign('small'), - normal_sign=Sign('normal'), - big_sign=Sign('big'), + small_sign=Sign("small"), + normal_sign=Sign("normal"), + big_sign=Sign("big"), ).first() self.assertIsInstance(obj.small_sign, int) self.assertIsInstance(obj.normal_sign, int) @@ -47,7 +50,7 @@ class SignTests(TestCase): def test_transform(self): with register_lookup(DecimalField, Sign): - DecimalModel.objects.create(n1=Decimal('5.4'), n2=Decimal('0')) - DecimalModel.objects.create(n1=Decimal('-0.1'), n2=Decimal('0')) + DecimalModel.objects.create(n1=Decimal("5.4"), n2=Decimal("0")) + DecimalModel.objects.create(n1=Decimal("-0.1"), n2=Decimal("0")) obj = DecimalModel.objects.filter(n1__sign__lt=0, n2__sign=0).get() - self.assertEqual(obj.n1, Decimal('-0.1')) + self.assertEqual(obj.n1, Decimal("-0.1")) |