Fix threading tests

Author: saulshanabrook

python/tests/test_bindings.py

@@ -1,4 +1,4 @@
-import _thread
+import concurrent.futures
 import json
 import os
 import pathlib
@@ -172,23 +172,6 @@ def test_sort_alias(self):
             Extract(DUMMY_SPAN, Var(DUMMY_SPAN, "my_map2"), Lit(DUMMY_SPAN, Int(0))),
         )
 
-        extract_report = egraph.extract_report()
-        assert isinstance(extract_report, Best)
-        assert extract_report.termdag.term_to_expr(extract_report.term, DUMMY_SPAN) == Call(
-            DUMMY_SPAN,
-            "map-insert",
-            [
-                Call(
-                    DUMMY_SPAN,
-                    "map-insert",
-                    [Call(DUMMY_SPAN, "map-empty", []), Lit(DUMMY_SPAN, Int(2)), Lit(DUMMY_SPAN, String("two"))],
-                ),
-                Lit(DUMMY_SPAN, Int(1)),
-                Lit(DUMMY_SPAN, String("one")),
-            ],
-        )
-        assert extract_report.cost == 4
-
 
 class TestVariant:
     def test_repr(self):
@@ -228,16 +211,18 @@ def test_cmds(self):
             ),
             RunSchedule(Repeat(DUMMY_SPAN, 10, Run(DUMMY_SPAN, RunConfig("")))),
         )
-
-        _thread.start_new_thread(print, cmds)
+        with concurrent.futures.ThreadPoolExecutor(max_workers=1) as executor:
+            executor.submit(print, cmds).result()
 
     @pytest.mark.xfail(reason="egraphs are unsendable")
     def test_egraph(self):
-        _thread.start_new_thread(
-            EGraph().run_program, (Datatype(DUMMY_SPAN, "Math", [Variant(DUMMY_SPAN, "Add", ["Math", "Math"])]),)
-        )
+        with concurrent.futures.ThreadPoolExecutor(max_workers=1) as executor:
+            executor.submit(
+                EGraph().run_program, Datatype(DUMMY_SPAN, "Math", [Variant(DUMMY_SPAN, "Add", ["Math", "Math"])])
+            ).result()
 
     def test_serialized_egraph(self):
         egraph = EGraph()
         serialized = egraph.serialize([])
-        _thread.start_new_thread(print, (serialized,))
+        with concurrent.futures.ThreadPoolExecutor(max_workers=1) as executor:
+            executor.submit(print, (serialized,)).result()