1import math
2from unittest import TestCase
3from simplejson.compat import long_type, text_type
4import simplejson as json
5from simplejson.decoder import NaN, PosInf, NegInf
6
7class TestFloat(TestCase):
8    def test_degenerates_allow(self):
9        for inf in (PosInf, NegInf):
10            self.assertEqual(json.loads(json.dumps(inf)), inf)
11        # Python 2.5 doesn't have math.isnan
12        nan = json.loads(json.dumps(NaN))
13        self.assertTrue((0 + nan) != nan)
14
15    def test_degenerates_ignore(self):
16        for f in (PosInf, NegInf, NaN):
17            self.assertEqual(json.loads(json.dumps(f, ignore_nan=True)), None)
18
19    def test_degenerates_deny(self):
20        for f in (PosInf, NegInf, NaN):
21            self.assertRaises(ValueError, json.dumps, f, allow_nan=False)
22
23    def test_floats(self):
24        for num in [1617161771.7650001, math.pi, math.pi**100,
25                    math.pi**-100, 3.1]:
26            self.assertEqual(float(json.dumps(num)), num)
27            self.assertEqual(json.loads(json.dumps(num)), num)
28            self.assertEqual(json.loads(text_type(json.dumps(num))), num)
29
30    def test_ints(self):
31        for num in [1, long_type(1), 1<<32, 1<<64]:
32            self.assertEqual(json.dumps(num), str(num))
33            self.assertEqual(int(json.dumps(num)), num)
34            self.assertEqual(json.loads(json.dumps(num)), num)
35            self.assertEqual(json.loads(text_type(json.dumps(num))), num)
36