44import cloudpickle
55import zmq
66
7- from executorlib .backend .interactive_serial import main
7+ from executorlib .backend .interactive_serial import evaluate_cmd
88
99
1010def calc (i , j ):
1111 return i + j
1212
1313
14- def set_global ():
14+ def set_global_variable ():
1515 return {"j" : 5 }
1616
1717
@@ -21,15 +21,15 @@ def raise_error():
2121
2222def submit (socket ):
2323 socket .send (
24- cloudpickle .dumps ({"init" : True , "fn" : set_global , "args" : (), "kwargs" : {}})
24+ cloudpickle .dumps ({"init" : True , "fn" : set_global_variable , "args" : (), "kwargs" : {}})
2525 )
2626 socket .send (cloudpickle .dumps ({"fn" : calc , "args" : (), "kwargs" : {"i" : 2 }}))
2727 socket .send (cloudpickle .dumps ({"shutdown" : True , "wait" : True }))
2828
2929
3030def submit_error (socket ):
3131 socket .send (
32- cloudpickle .dumps ({"init" : True , "fn" : set_global , "args" : (), "kwargs" : {}})
32+ cloudpickle .dumps ({"init" : True , "fn" : set_global_variable , "args" : (), "kwargs" : {}})
3333 )
3434 socket .send (cloudpickle .dumps ({"fn" : calc , "args" : (), "kwargs" : {}}))
3535 socket .send (cloudpickle .dumps ({"shutdown" : True , "wait" : True }))
@@ -48,7 +48,7 @@ def test_main_as_thread(self):
4848 context = zmq .Context ()
4949 socket = context .socket (zmq .PAIR )
5050 port = socket .bind_to_random_port ("tcp://*" )
51- t = Thread (target = main , kwargs = {"argument_lst" : ["--zmqport" , str (port )]})
51+ t = Thread (target = evaluate_cmd , kwargs = {"argument_lst" : ["--zmqport" , str (port )]})
5252 t .start ()
5353 submit (socket = socket )
5454 self .assertEqual (cloudpickle .loads (socket .recv ()), {"result" : True })
@@ -61,7 +61,7 @@ def test_main_as_thread_error(self):
6161 context = zmq .Context ()
6262 socket = context .socket (zmq .PAIR )
6363 port = socket .bind_to_random_port ("tcp://*" )
64- t = Thread (target = main , kwargs = {"argument_lst" : ["--zmqport" , str (port )]})
64+ t = Thread (target = evaluate_cmd , kwargs = {"argument_lst" : ["--zmqport" , str (port )]})
6565 t .start ()
6666 submit_error (socket = socket )
6767 self .assertEqual (cloudpickle .loads (socket .recv ()), {"result" : True })
@@ -76,7 +76,7 @@ def test_main_as_thread_init_error(self):
7676 context = zmq .Context ()
7777 socket = context .socket (zmq .PAIR )
7878 port = socket .bind_to_random_port ("tcp://*" )
79- t = Thread (target = main , kwargs = {"argument_lst" : ["--zmqport" , str (port )]})
79+ t = Thread (target = evaluate_cmd , kwargs = {"argument_lst" : ["--zmqport" , str (port )]})
8080 t .start ()
8181 submit_init_error (socket = socket )
8282 self .assertEqual (
@@ -94,7 +94,7 @@ def test_submit_as_thread(self):
9494 port = socket .bind_to_random_port ("tcp://*" )
9595 t = Thread (target = submit , kwargs = {"socket" : socket })
9696 t .start ()
97- main (argument_lst = ["--zmqport" , str (port )])
97+ evaluate_cmd (argument_lst = ["--zmqport" , str (port )])
9898 self .assertEqual (cloudpickle .loads (socket .recv ()), {"result" : True })
9999 self .assertEqual (cloudpickle .loads (socket .recv ()), {"result" : 7 })
100100 self .assertEqual (cloudpickle .loads (socket .recv ()), {"result" : True })
@@ -107,7 +107,7 @@ def test_submit_as_thread_error(self):
107107 port = socket .bind_to_random_port ("tcp://*" )
108108 t = Thread (target = submit_error , kwargs = {"socket" : socket })
109109 t .start ()
110- main (argument_lst = ["--zmqport" , str (port )])
110+ evaluate_cmd (argument_lst = ["--zmqport" , str (port )])
111111 self .assertEqual (cloudpickle .loads (socket .recv ()), {"result" : True })
112112 self .assertEqual (
113113 str (type (cloudpickle .loads (socket .recv ())["error" ])), "<class 'TypeError'>"
0 commit comments