39 lines
1.6 KiB
Python
39 lines
1.6 KiB
Python
import math
|
|
from unittest import TestCase
|
|
from simplejson.compat import long_type, text_type
|
|
import simplejson as json
|
|
from simplejson.decoder import NaN, PosInf, NegInf
|
|
|
|
class TestFloat(TestCase):
|
|
def test_degenerates_allow(self):
|
|
for inf in (PosInf, NegInf):
|
|
self.assertEqual(json.loads(json.dumps(inf, allow_nan=True), allow_nan=True), inf)
|
|
# Python 2.5 doesn't have math.isnan
|
|
nan = json.loads(json.dumps(NaN, allow_nan=True), allow_nan=True)
|
|
self.assertTrue((0 + nan) != nan)
|
|
|
|
def test_degenerates_ignore(self):
|
|
for f in (PosInf, NegInf, NaN):
|
|
self.assertEqual(json.loads(json.dumps(f, ignore_nan=True)), None)
|
|
|
|
def test_degenerates_deny(self):
|
|
for f in (PosInf, NegInf, NaN):
|
|
self.assertRaises(ValueError, json.dumps, f, allow_nan=False)
|
|
for s in ('Infinity', '-Infinity', 'NaN'):
|
|
self.assertRaises(ValueError, json.loads, s, allow_nan=False)
|
|
self.assertRaises(ValueError, json.loads, s)
|
|
|
|
def test_floats(self):
|
|
for num in [1617161771.7650001, math.pi, math.pi**100,
|
|
math.pi**-100, 3.1]:
|
|
self.assertEqual(float(json.dumps(num)), num)
|
|
self.assertEqual(json.loads(json.dumps(num)), num)
|
|
self.assertEqual(json.loads(text_type(json.dumps(num))), num)
|
|
|
|
def test_ints(self):
|
|
for num in [1, long_type(1), 1<<32, 1<<64]:
|
|
self.assertEqual(json.dumps(num), str(num))
|
|
self.assertEqual(int(json.dumps(num)), num)
|
|
self.assertEqual(json.loads(json.dumps(num)), num)
|
|
self.assertEqual(json.loads(text_type(json.dumps(num))), num)
|