17db96d56Sopenharmony_ciimport math
27db96d56Sopenharmony_cifrom test.test_json import PyTest, CTest
37db96d56Sopenharmony_ci
47db96d56Sopenharmony_ci
57db96d56Sopenharmony_ciclass TestFloat:
67db96d56Sopenharmony_ci    def test_floats(self):
77db96d56Sopenharmony_ci        for num in [1617161771.7650001, math.pi, math.pi**100, math.pi**-100, 3.1]:
87db96d56Sopenharmony_ci            self.assertEqual(float(self.dumps(num)), num)
97db96d56Sopenharmony_ci            self.assertEqual(self.loads(self.dumps(num)), num)
107db96d56Sopenharmony_ci
117db96d56Sopenharmony_ci    def test_ints(self):
127db96d56Sopenharmony_ci        for num in [1, 1<<32, 1<<64]:
137db96d56Sopenharmony_ci            self.assertEqual(self.dumps(num), str(num))
147db96d56Sopenharmony_ci            self.assertEqual(int(self.dumps(num)), num)
157db96d56Sopenharmony_ci
167db96d56Sopenharmony_ci    def test_out_of_range(self):
177db96d56Sopenharmony_ci        self.assertEqual(self.loads('[23456789012E666]'), [float('inf')])
187db96d56Sopenharmony_ci        self.assertEqual(self.loads('[-23456789012E666]'), [float('-inf')])
197db96d56Sopenharmony_ci
207db96d56Sopenharmony_ci    def test_allow_nan(self):
217db96d56Sopenharmony_ci        for val in (float('inf'), float('-inf'), float('nan')):
227db96d56Sopenharmony_ci            out = self.dumps([val])
237db96d56Sopenharmony_ci            if val == val:  # inf
247db96d56Sopenharmony_ci                self.assertEqual(self.loads(out), [val])
257db96d56Sopenharmony_ci            else:  # nan
267db96d56Sopenharmony_ci                res = self.loads(out)
277db96d56Sopenharmony_ci                self.assertEqual(len(res), 1)
287db96d56Sopenharmony_ci                self.assertNotEqual(res[0], res[0])
297db96d56Sopenharmony_ci            self.assertRaises(ValueError, self.dumps, [val], allow_nan=False)
307db96d56Sopenharmony_ci
317db96d56Sopenharmony_ci
327db96d56Sopenharmony_ciclass TestPyFloat(TestFloat, PyTest): pass
337db96d56Sopenharmony_ciclass TestCFloat(TestFloat, CTest): pass
34