Exemplo n.º 1
0
    async def test_imports_not_used(self):
        c = Config()
        c.PooledKernelManager.kernel_pools = {NATIVE_KERNEL_NAME: 1}
        c.PooledKernelManager.strict_pool_names = True
        c.python_imports = []
        km = PooledKernelManager(config=c)

        try:
            kid = await km.start_kernel()
            kernel = km.get_kernel(kid)
            self.assertIsNotNone(kernel)

            client = ExecClient(kernel, _store_outputs=True)
            async with client.setup_kernel():
                await client.execute(
                    'import sys\nprint("turtle" in sys.modules)')
            self.assertEqual(
                client._outputs,
                [{
                    "name": "stdout",
                    "output_type": "stream",
                    "text": "False\n"
                }],
            )
        finally:
            await km.shutdown_all()