test: reorgainize test files

Author: XuehaiPan

tests/concurrent/test_subinterpreters.py

@@ -0,0 +1,14 @@
+# Copyright 2022-2025 MetaOPT Team. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================

tests/test_concurrent.py tests/concurrent/test_threading.pytests/test_concurrent.py renamed to tests/concurrent/test_threading.py

@@ -55,8 +55,8 @@
 atexit.register(EXECUTOR.shutdown)
 
 
-def concurrent_run(func):
-    futures = [EXECUTOR.submit(func) for _ in range(NUM_FUTURES)]
+def concurrent_run(func, /, *args, **kwargs):
+    futures = [EXECUTOR.submit(func, *args, **kwargs) for _ in range(NUM_FUTURES)]
     future2index = {future: i for i, future in enumerate(futures)}
     completed_futures = sorted(as_completed(futures), key=future2index.get)
     first_exception = next(filter(None, (future.exception() for future in completed_futures)), None)
@@ -92,7 +92,7 @@ def test_fn():
         for result in concurrent_run(test_fn):
             assert result == expected
 
-    for result in concurrent_run(lambda: optree.tree_unflatten(treespec, leaves)):
+    for result in concurrent_run(optree.tree_unflatten, treespec, leaves):
         assert result == tree
 
 
@@ -353,7 +353,7 @@ def test_tree_iter_thread_safe(
             namespace=namespace,
         )
 
-    results = concurrent_run(lambda: list(it))
+    results = concurrent_run(list, it)
     for seq in results:
         assert sorted(seq) == seq
     assert sorted(itertools.chain.from_iterable(results)) == list(range(num_leaves))