Skip to content

Commit 04a66a8

Browse files
committed
Issue MagicStack#1020 Infinity numeric support
1 parent 89d5bd0 commit 04a66a8

File tree

2 files changed

+5
-11
lines changed

2 files changed

+5
-11
lines changed

tests/test_codecs.py

+4-10
Original file line numberDiff line numberDiff line change
@@ -633,17 +633,11 @@ async def test_numeric(self):
633633
"SELECT $1::numeric", decimal.Decimal('sNaN'))
634634
self.assertTrue(res.is_nan())
635635

636-
with self.assertRaisesRegex(asyncpg.DataError,
637-
'numeric type does not '
638-
'support infinite values'):
639-
await self.con.fetchval(
640-
"SELECT $1::numeric", decimal.Decimal('-Inf'))
636+
res = await self.con.fetchval("SELECT $1::numeric", decimal.Decimal("-Inf"))
637+
self.assertTrue(res.is_infinite())
641638

642-
with self.assertRaisesRegex(asyncpg.DataError,
643-
'numeric type does not '
644-
'support infinite values'):
645-
await self.con.fetchval(
646-
"SELECT $1::numeric", decimal.Decimal('+Inf'))
639+
res = await self.con.fetchval("SELECT $1::numeric", decimal.Decimal("+Inf"))
640+
self.assertTrue(res.is_infinite())
647641

648642
with self.assertRaisesRegex(asyncpg.DataError, 'invalid'):
649643
await self.con.fetchval(

0 commit comments

Comments
 (0)