diff options
Diffstat (limited to 'tests/db_functions/math/test_cos.py')
| -rw-r--r-- | tests/db_functions/math/test_cos.py | 21 |
1 files changed, 10 insertions, 11 deletions
diff --git a/tests/db_functions/math/test_cos.py b/tests/db_functions/math/test_cos.py index 73e59f0463..9e96589bac 100644 --- a/tests/db_functions/math/test_cos.py +++ b/tests/db_functions/math/test_cos.py @@ -10,15 +10,14 @@ from ..models import DecimalModel, FloatModel, IntegerModel class CosTests(TestCase): - def test_null(self): IntegerModel.objects.create() - obj = IntegerModel.objects.annotate(null_cos=Cos('normal')).first() + obj = IntegerModel.objects.annotate(null_cos=Cos("normal")).first() self.assertIsNone(obj.null_cos) def test_decimal(self): - DecimalModel.objects.create(n1=Decimal('-12.9'), n2=Decimal('0.6')) - obj = DecimalModel.objects.annotate(n1_cos=Cos('n1'), n2_cos=Cos('n2')).first() + DecimalModel.objects.create(n1=Decimal("-12.9"), n2=Decimal("0.6")) + obj = DecimalModel.objects.annotate(n1_cos=Cos("n1"), n2_cos=Cos("n2")).first() self.assertIsInstance(obj.n1_cos, Decimal) self.assertIsInstance(obj.n2_cos, Decimal) self.assertAlmostEqual(obj.n1_cos, Decimal(math.cos(obj.n1))) @@ -26,7 +25,7 @@ class CosTests(TestCase): def test_float(self): FloatModel.objects.create(f1=-27.5, f2=0.33) - obj = FloatModel.objects.annotate(f1_cos=Cos('f1'), f2_cos=Cos('f2')).first() + obj = FloatModel.objects.annotate(f1_cos=Cos("f1"), f2_cos=Cos("f2")).first() self.assertIsInstance(obj.f1_cos, float) self.assertIsInstance(obj.f2_cos, float) self.assertAlmostEqual(obj.f1_cos, math.cos(obj.f1)) @@ -35,9 +34,9 @@ class CosTests(TestCase): def test_integer(self): IntegerModel.objects.create(small=-20, normal=15, big=-1) obj = IntegerModel.objects.annotate( - small_cos=Cos('small'), - normal_cos=Cos('normal'), - big_cos=Cos('big'), + small_cos=Cos("small"), + normal_cos=Cos("normal"), + big_cos=Cos("big"), ).first() self.assertIsInstance(obj.small_cos, float) self.assertIsInstance(obj.normal_cos, float) @@ -48,7 +47,7 @@ class CosTests(TestCase): def test_transform(self): with register_lookup(DecimalField, Cos): - DecimalModel.objects.create(n1=Decimal('-8.0'), n2=Decimal('0')) - DecimalModel.objects.create(n1=Decimal('3.14'), n2=Decimal('0')) + DecimalModel.objects.create(n1=Decimal("-8.0"), n2=Decimal("0")) + DecimalModel.objects.create(n1=Decimal("3.14"), n2=Decimal("0")) obj = DecimalModel.objects.filter(n1__cos__gt=-0.2).get() - self.assertEqual(obj.n1, Decimal('-8.0')) + self.assertEqual(obj.n1, Decimal("-8.0")) |