diff options
Diffstat (limited to 'tests/db_functions/math/test_cot.py')
| -rw-r--r-- | tests/db_functions/math/test_cot.py | 21 |
1 files changed, 10 insertions, 11 deletions
diff --git a/tests/db_functions/math/test_cot.py b/tests/db_functions/math/test_cot.py index d876648598..04fdcb475c 100644 --- a/tests/db_functions/math/test_cot.py +++ b/tests/db_functions/math/test_cot.py @@ -10,15 +10,14 @@ from ..models import DecimalModel, FloatModel, IntegerModel class CotTests(TestCase): - def test_null(self): IntegerModel.objects.create() - obj = IntegerModel.objects.annotate(null_cot=Cot('normal')).first() + obj = IntegerModel.objects.annotate(null_cot=Cot("normal")).first() self.assertIsNone(obj.null_cot) def test_decimal(self): - DecimalModel.objects.create(n1=Decimal('-12.9'), n2=Decimal('0.6')) - obj = DecimalModel.objects.annotate(n1_cot=Cot('n1'), n2_cot=Cot('n2')).first() + DecimalModel.objects.create(n1=Decimal("-12.9"), n2=Decimal("0.6")) + obj = DecimalModel.objects.annotate(n1_cot=Cot("n1"), n2_cot=Cot("n2")).first() self.assertIsInstance(obj.n1_cot, Decimal) self.assertIsInstance(obj.n2_cot, Decimal) self.assertAlmostEqual(obj.n1_cot, Decimal(1 / math.tan(obj.n1))) @@ -26,7 +25,7 @@ class CotTests(TestCase): def test_float(self): FloatModel.objects.create(f1=-27.5, f2=0.33) - obj = FloatModel.objects.annotate(f1_cot=Cot('f1'), f2_cot=Cot('f2')).first() + obj = FloatModel.objects.annotate(f1_cot=Cot("f1"), f2_cot=Cot("f2")).first() self.assertIsInstance(obj.f1_cot, float) self.assertIsInstance(obj.f2_cot, float) self.assertAlmostEqual(obj.f1_cot, 1 / math.tan(obj.f1)) @@ -35,9 +34,9 @@ class CotTests(TestCase): def test_integer(self): IntegerModel.objects.create(small=-5, normal=15, big=-1) obj = IntegerModel.objects.annotate( - small_cot=Cot('small'), - normal_cot=Cot('normal'), - big_cot=Cot('big'), + small_cot=Cot("small"), + normal_cot=Cot("normal"), + big_cot=Cot("big"), ).first() self.assertIsInstance(obj.small_cot, float) self.assertIsInstance(obj.normal_cot, float) @@ -48,7 +47,7 @@ class CotTests(TestCase): def test_transform(self): with register_lookup(DecimalField, Cot): - DecimalModel.objects.create(n1=Decimal('12.0'), n2=Decimal('0')) - DecimalModel.objects.create(n1=Decimal('1.0'), n2=Decimal('0')) + DecimalModel.objects.create(n1=Decimal("12.0"), n2=Decimal("0")) + DecimalModel.objects.create(n1=Decimal("1.0"), n2=Decimal("0")) obj = DecimalModel.objects.filter(n1__cot__gt=0).get() - self.assertEqual(obj.n1, Decimal('1.0')) + self.assertEqual(obj.n1, Decimal("1.0")) |