Add base class test

Author: jacobtomlinson

dask_jobqueue/tests/test_runner.py

@@ -0,0 +1,29 @@
+import asyncio
+from contextlib import suppress
+from time import time
+
+import pytest
+
+from dask.distributed import Client
+
+from dask_jobqueue.runner import AsyncCommWorld, AsyncRunner
+
+
+@pytest.mark.asyncio
+@pytest.mark.timeout(10)
+async def test_runner():
+    commworld = AsyncCommWorld()
+
+    async def run_code(commworld):
+        with suppress(SystemExit):
+            async with AsyncRunner(commworld, asynchronous=True) as runner:
+                async with Client(runner, asynchronous=True) as c:
+                    start = time()
+                    while len(c.scheduler_info()["workers"]) != 2:
+                        assert time() < start + 10
+                        await asyncio.sleep(0.2)
+
+                    assert await c.submit(lambda x: x + 1, 10).result() == 11
+                    assert await c.submit(lambda x: x + 1, 20).result() == 21
+
+    await asyncio.gather(*[run_code(commworld) for _ in range(4)])