format

Author: eendebakpt

Lib/test/test_free_threading/test_json.py

@@ -4,11 +4,15 @@
 from test.support import threading_helper
 
 
-def encode_json_helper(json, worker, data, number_of_threads, number_of_json_encodings=100):
+def encode_json_helper(
+    json, worker, data, number_of_threads=12, number_of_json_encodings=100
+):
     worker_threads = []
     barrier = Barrier(number_of_threads)
     for index in range(number_of_threads):
-        worker_threads.append(Thread(target=worker, args=[barrier, data, index]))
+        worker_threads.append(
+            Thread(target=worker, args=[barrier, data, index])
+        )
     for t in worker_threads:
         t.start()
     for ii in range(number_of_json_encodings):
@@ -33,7 +37,6 @@ class TestJsonEncoding(CTest):
     # corrupt the interpreter
 
     def test_json_mutating_list(self):
-
         def worker(barrier, data, index):
             barrier.wait()
             while data:
@@ -42,28 +45,28 @@ def worker(barrier, data, index):
                         d.clear()
                     else:
                         d.append(index)
-                        d.append(index)
-                        d.append(index)
-        encode_json_helper(self.json, worker,  [[], []], number_of_threads=16)
 
-    def test_json_mutating_dict(self):
+        data = [[], []]
+        encode_json_helper(self.json, worker, data)
 
+    def test_json_mutating_exact_dict(self):
         def worker(barrier, data, index):
             barrier.wait()
             while data:
                 for d in data:
                     if len(d) > 5:
                         try:
-                            d.pop(list(d)[0])
+                            key = list(d)[0]
+                            d.pop()
                         except (KeyError, IndexError):
                             pass
                     else:
                         d[index] = index
 
-        encode_json_helper(self.json, worker, [{}, {}], number_of_threads=16)
+        data = [{}, {}]
+        encode_json_helper(self.json, worker, data)
 
     def test_json_mutating_mapping(self):
-
         def worker(barrier, data, index):
             barrier.wait()
             while data:
@@ -74,13 +77,12 @@ def worker(barrier, data, index):
                         d.mapping.append((index, index))
 
         data = [MyMapping(), MyMapping()]
-        encode_json_helper(self.json, worker, data, number_of_threads=16)
+        encode_json_helper(self.json, worker, data)
 
 
 if __name__ == "__main__":
     import time
-
     t0 = time.time()
     unittest.main()
-    dt = time.time()-t0
-    print(f'Done: {dt:.2f}')
+    dt = time.time() - t0
+    print(f"Done: {dt:.2f}")