a ì<+bdã@s†ddlmZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZdd „ZGd d „d ejƒZdS) é)ÚStringION)ÚBlockingInProcessKernelClient)ÚInProcessKernelManager)ÚInProcessKernel)Úassemble_output)Úcapture_outputcCsptj d¡rltjdkrltjdkrlddl}zddlm}m}WntyNYn0t |  ¡ƒ|url|  |ƒ¡dS)aset default asyncio policy to be compatible with tornado Tornado 6 (at least) is not compatible with the default asyncio implementation on Windows Pick the older SelectorEventLoopPolicy on Windows if the known-incompatible default policy is in use. do this as early as possible to make it a low priority and overrideable ref: https://github.com/tornadoweb/tornado/issues/2608 FIXME: if/when tornado supports the defaults in asyncio, remove and bump tornado requirement for py38 Úwin)éé)éérN)ÚWindowsProactorEventLoopPolicyÚWindowsSelectorEventLoopPolicy) ÚsysÚplatformÚ startswithÚ version_infoÚtornadoÚasyncior rÚ ImportErrorÚtypeZget_event_loop_policyZset_event_loop_policy)rr r©rúDlib/python3.9/site-packages/ipykernel/inprocess/tests/test_kernel.pyÚ_init_asyncio_patchs  rc@s`eZdZdd„Zdd„Zdd„Zejjde j vdd d d „ƒZ ejj d d d d„ƒZ dd„ZdS)ÚInProcessKernelTestCasecCs<tƒtƒ|_|j ¡|j ¡|_|j ¡|j ¡dS©N)rrÚkmZ start_kernelZclientÚkcZstart_channelsZwait_for_ready)ÚselfrrrÚsetUp7s    zInProcessKernelTestCase.setUpcCs<tjddd}|j}| d¡t|jƒ\}}| d|¡dS)z*Does %pylab work in the in-process kernel?Ú matplotlibzThis test requires matplotlib©Úreasonz%pylabN)ÚpytestZ importorskiprÚexecuterÚ get_iopub_msgZassertIn)rr rÚoutÚerrrrrÚ test_pylab?s  z"InProcessKernelTestCase.test_pylabcCsRtdƒ}tj}|t_z|j d¡W|t_n|t_0|jjjj  d¡dksNJ‚dS)z@ Does the in-process kernel handle raw_input correctly? zfoobar z x = input()ÚxZfoobarN) rrÚstdinrr$rÚkernelÚshellZuser_nsÚget)rÚioZ sys_stdinrrrÚtest_raw_inputGsz&InProcessKernelTestCase.test_raw_inputZ__pypy__z fails on pypyr!cCsˆtƒ}tƒ}|j d¡Wdƒn1s.0Y|jdksFJ‚t||jd}|j |¡|  d¡t |j ƒ\}}|dks„J‚dS)z: Does the in-process kernel correctly capture IO? ú print("foo")Núfoo ©r+Úsessionz print("bar")zbar ) rrr,Úrun_cellÚstdoutrr3Ú frontendsÚappendr$rr%©rr+r.rr&r'rrrÚ test_stdoutSs*  z#InProcessKernelTestCase.test_stdoutzDCurrently don't capture during test as pytest does its own capturingcCs’tƒ}tƒ}|j d¡Wdƒn1s.0Y|jdksFJ‚t||jd}|j |¡|  d¡|  d¡t |j ƒ\}}|dksŽJ‚dS)zDoes correctly capture fdr0Nr1r2z import oszos.system("echo capfd")zcapfd ) rrr,r4r5rr3r6r7r$rZ iopub_channelr8rrrÚ test_capfdfs*   z"InProcessKernelTestCase.test_capfdcCs&tƒ}d|_dd„|_|jdddS)z5Tests that kernel getpass accept the stream parameterTc_sdSrr)ÚargsÚkwargsrrrÚ|óz=InProcessKernelTestCase.test_getpass_stream..z non empty)ÚstreamN)rZ _allow_stdinZ_input_requestZgetpass)rr+rrrÚtest_getpass_streamxs z+InProcessKernelTestCase.test_getpass_streamN)Ú__name__Ú __module__Ú __qualname__rr(r/r#ZmarkZskipifrÚbuiltin_module_namesr9Úskipr:r@rrrrr5s þ ÿ r)r.rrZunittestr#rZipykernel.inprocess.blockingrZipykernel.inprocess.managerrZipykernel.inprocess.ipkernelrZipykernel.tests.utilsrZIPython.utils.iorrZTestCaserrrrrÚs      !