Skip to content

Commit 4e2e033

Browse files
committed
Merge remote-tracking branch 'origin/remove_shell' into remove_shell
2 parents befe209 + 8203a3d commit 4e2e033

File tree

2 files changed

+40
-13
lines changed

2 files changed

+40
-13
lines changed

tests/test_shell_executor.py

Lines changed: 33 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,12 @@
88
from executorlib.shared.spawner import MpiExecSpawner
99

1010

11-
def submit_shell_command(command: list, universal_newlines: bool = True, shell: bool = False):
12-
return subprocess.check_output(command, universal_newlines=universal_newlines, shell=shell)
11+
def submit_shell_command(
12+
command: list, universal_newlines: bool = True, shell: bool = False
13+
):
14+
return subprocess.check_output(
15+
command, universal_newlines=universal_newlines, shell=shell
16+
)
1317

1418

1519
class SubprocessExecutorTest(unittest.TestCase):
@@ -77,31 +81,51 @@ def test_broken_executable(self):
7781

7882
def test_shell_static_executor_args(self):
7983
with Executor(max_workers=1) as exe:
80-
future = exe.submit(submit_shell_command, ["echo", "test"], universal_newlines=True, shell=False)
84+
future = exe.submit(
85+
submit_shell_command,
86+
["echo", "test"],
87+
universal_newlines=True,
88+
shell=False,
89+
)
8190
self.assertFalse(future.done())
8291
self.assertEqual("test\n", future.result())
8392
self.assertTrue(future.done())
8493

8594
def test_shell_static_executor_binary(self):
8695
with Executor(max_workers=1) as exe:
87-
future = exe.submit(submit_shell_command, ["echo", "test"], universal_newlines=False, shell=False)
96+
future = exe.submit(
97+
submit_shell_command,
98+
["echo", "test"],
99+
universal_newlines=False,
100+
shell=False,
101+
)
88102
self.assertFalse(future.done())
89103
self.assertEqual(b"test\n", future.result())
90104
self.assertTrue(future.done())
91105

92106
def test_shell_static_executor_shell(self):
93107
with Executor(max_workers=1) as exe:
94-
future = exe.submit(submit_shell_command, "echo test", universal_newlines=True, shell=True)
108+
future = exe.submit(
109+
submit_shell_command, "echo test", universal_newlines=True, shell=True
110+
)
95111
self.assertFalse(future.done())
96112
self.assertEqual("test\n", future.result())
97113
self.assertTrue(future.done())
98114

99115
def test_shell_executor(self):
100116
with Executor(max_workers=2) as exe:
101-
f_1 = exe.submit(submit_shell_command, ["echo", "test_1"], universal_newlines=True)
102-
f_2 = exe.submit(submit_shell_command, ["echo", "test_2"], universal_newlines=True)
103-
f_3 = exe.submit(submit_shell_command, ["echo", "test_3"], universal_newlines=True)
104-
f_4 = exe.submit(submit_shell_command, ["echo", "test_4"], universal_newlines=True)
117+
f_1 = exe.submit(
118+
submit_shell_command, ["echo", "test_1"], universal_newlines=True
119+
)
120+
f_2 = exe.submit(
121+
submit_shell_command, ["echo", "test_2"], universal_newlines=True
122+
)
123+
f_3 = exe.submit(
124+
submit_shell_command, ["echo", "test_3"], universal_newlines=True
125+
)
126+
f_4 = exe.submit(
127+
submit_shell_command, ["echo", "test_4"], universal_newlines=True
128+
)
105129
self.assertFalse(f_1.done())
106130
self.assertFalse(f_2.done())
107131
self.assertFalse(f_3.done())

tests/test_shell_interactive.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ def init_process():
1919
stdin=subprocess.PIPE,
2020
stdout=subprocess.PIPE,
2121
universal_newlines=True,
22-
shell=False
22+
shell=False,
2323
)
2424
}
2525

@@ -102,15 +102,18 @@ def test_execute_single_task(self):
102102
test_queue.join()
103103

104104
def test_shell_interactive_executor(self):
105-
with (Executor(
105+
with Executor(
106106
init_function=init_process,
107107
block_allocation=True,
108-
) as exe):
108+
) as exe:
109109
future_lines = exe.submit(
110110
interact, shell_input="4\n", lines_to_read=5, stop_read_pattern=None
111111
)
112112
future_pattern = exe.submit(
113-
interact, shell_input="4\n", lines_to_read=None, stop_read_pattern="done"
113+
interact,
114+
shell_input="4\n",
115+
lines_to_read=None,
116+
stop_read_pattern="done",
114117
)
115118
self.assertFalse(future_lines.done())
116119
self.assertFalse(future_pattern.done())

0 commit comments

Comments
 (0)