Add test to json.dump for newly added encode_float argument

Lee-W · Lee-W · commit 8a38261af21e · 2020-12-16T22:50:39.000+08:00
diff --git a/Lib/test/test_json/test_dump.py b/Lib/test/test_json/test_dump.py
@@ -57,6 +57,21 @@ def __lt__(self, o):
         d[1337] = "true.dat"
         self.assertEqual(self.dumps(d, sort_keys=True), '{"1337": "true.dat"}')
 
+    # Issue 36841
+    def test_encode_float(self):
+        data = {0.88: 0.9, 0: 0.1}
+        expected = '{"0.88": 0.9, "0": 0.1}'
+        self.assertEqual(
+            self.dumps(data, encode_float=None),
+            expected
+        )
+
+        data = {0.88: 0.9, 0: 0.1}
+        expected = '{"1": 1, "0": 0}'
+        self.assertEqual(
+            self.dumps(data, encode_float=lambda x: str(round(x))),
+            expected
+        )
 
 class TestPyDump(TestDump, PyTest): pass
 
