47 lines
2.0 KiB
Python
47 lines
2.0 KiB
Python
import sys
|
|
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)
|
|
|
|
def test_float_range(self):
|
|
try:
|
|
float_range = [sys.float_info.min, sys.float_info.max]
|
|
except AttributeError:
|
|
float_range = [2.2250738585072014e-308, 1.7976931348623157e+308]
|
|
self.assertEqual(json.loads(json.dumps(float_range)), float_range)
|