Skip to content

Commit d8f82a3

Browse files
committed
Add test for local dask
1 parent d939168 commit d8f82a3

1 file changed

Lines changed: 13 additions & 0 deletions

File tree

tests/test_moldrug.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,19 @@ def test_single_receptor_command_line():
158158
print(p.stdout)
159159

160160

161+
def test_dask_local():
162+
from moldrug.runner import RunnerMode, Runner
163+
164+
runner_instance = Runner(
165+
mode=RunnerMode.DASK_LOCAL,
166+
process_count=6
167+
)
168+
169+
result = runner_instance.run(sum, entries=[[1, 2], [2, 3]])
170+
assert result == [3, 5]
171+
runner_instance.client.close()
172+
173+
161174
def test_multi_receptor(maxiter=1, popsize=2, njobs=3, NumbCalls=1):
162175
out = utils.GA(
163176
seed_mol=[Chem.MolFromSmiles(TEST_DATA['x0161']['smiles']), Chem.MolFromSmiles(TEST_DATA['x0161']['smiles'])],

0 commit comments

Comments
 (0)