@@ -298,13 +298,15 @@ def test_no_cleanup_shared_context(self, zmq_context):
298
298
299
299
def test_subclass_callables (self , km_subclass ):
300
300
km_subclass .reset_counts ()
301
+
301
302
km_subclass .start_kernel (stdout = PIPE , stderr = PIPE )
302
303
assert km_subclass .call_count ("start_kernel" ) == 1
303
304
assert km_subclass .call_count ("_launch_kernel" ) == 1
304
305
305
306
is_alive = km_subclass .is_alive ()
306
307
assert is_alive
307
308
km_subclass .reset_counts ()
309
+
308
310
km_subclass .restart_kernel (now = True )
309
311
assert km_subclass .call_count ("restart_kernel" ) == 1
310
312
assert km_subclass .call_count ("shutdown_kernel" ) == 1
@@ -313,26 +315,32 @@ def test_subclass_callables(self, km_subclass):
313
315
assert km_subclass .call_count ("cleanup_resources" ) == 1
314
316
assert km_subclass .call_count ("start_kernel" ) == 1
315
317
assert km_subclass .call_count ("_launch_kernel" ) == 1
318
+ assert km_subclass .call_count ("signal_kernel" ) == 1
316
319
317
320
is_alive = km_subclass .is_alive ()
318
321
assert is_alive
319
-
322
+ assert km_subclass . call_count ( "is_alive" ) >= 1
320
323
km_subclass .reset_counts ()
324
+
321
325
km_subclass .interrupt_kernel ()
322
326
assert km_subclass .call_count ("interrupt_kernel" ) == 1
327
+ assert km_subclass .call_count ("signal_kernel" ) == 1
323
328
324
329
assert isinstance (km_subclass , KernelManager )
325
-
326
330
km_subclass .reset_counts ()
331
+
327
332
km_subclass .shutdown_kernel (now = False )
328
333
assert km_subclass .call_count ("shutdown_kernel" ) == 1
329
334
assert km_subclass .call_count ("interrupt_kernel" ) == 1
330
335
assert km_subclass .call_count ("request_shutdown" ) == 1
331
336
assert km_subclass .call_count ("finish_shutdown" ) == 1
332
337
assert km_subclass .call_count ("cleanup_resources" ) == 1
338
+ assert km_subclass .call_count ("signal_kernel" ) == 1
339
+ assert km_subclass .call_count ("is_alive" ) >= 1
333
340
334
341
is_alive = km_subclass .is_alive ()
335
342
assert is_alive is False
343
+ assert km_subclass .call_count ("is_alive" ) >= 1
336
344
assert km_subclass .context .closed
337
345
338
346
@@ -516,13 +524,16 @@ async def test_start_new_async_kernel(self, install_kernel, start_async_kernel):
516
524
517
525
async def test_subclass_callables (self , async_km_subclass ):
518
526
async_km_subclass .reset_counts ()
527
+
519
528
await async_km_subclass .start_kernel (stdout = PIPE , stderr = PIPE )
520
529
assert async_km_subclass .call_count ("start_kernel" ) == 1
521
530
assert async_km_subclass .call_count ("_launch_kernel" ) == 1
522
531
523
532
is_alive = await async_km_subclass .is_alive ()
524
533
assert is_alive
534
+ assert async_km_subclass .call_count ("is_alive" ) >= 1
525
535
async_km_subclass .reset_counts ()
536
+
526
537
await async_km_subclass .restart_kernel (now = True )
527
538
assert async_km_subclass .call_count ("restart_kernel" ) == 1
528
539
assert async_km_subclass .call_count ("shutdown_kernel" ) == 1
@@ -531,24 +542,30 @@ async def test_subclass_callables(self, async_km_subclass):
531
542
assert async_km_subclass .call_count ("cleanup_resources" ) == 1
532
543
assert async_km_subclass .call_count ("start_kernel" ) == 1
533
544
assert async_km_subclass .call_count ("_launch_kernel" ) == 1
545
+ assert async_km_subclass .call_count ("signal_kernel" ) == 1
534
546
535
547
is_alive = await async_km_subclass .is_alive ()
536
548
assert is_alive
537
-
549
+ assert async_km_subclass . call_count ( "is_alive" ) >= 1
538
550
async_km_subclass .reset_counts ()
551
+
539
552
await async_km_subclass .interrupt_kernel ()
540
553
assert async_km_subclass .call_count ("interrupt_kernel" ) == 1
554
+ assert async_km_subclass .call_count ("signal_kernel" ) == 1
541
555
542
556
assert isinstance (async_km_subclass , AsyncKernelManager )
543
-
544
557
async_km_subclass .reset_counts ()
558
+
545
559
await async_km_subclass .shutdown_kernel (now = False )
546
560
assert async_km_subclass .call_count ("shutdown_kernel" ) == 1
547
561
assert async_km_subclass .call_count ("interrupt_kernel" ) == 1
548
562
assert async_km_subclass .call_count ("request_shutdown" ) == 1
549
563
assert async_km_subclass .call_count ("finish_shutdown" ) == 1
550
564
assert async_km_subclass .call_count ("cleanup_resources" ) == 1
565
+ assert async_km_subclass .call_count ("signal_kernel" ) == 1
566
+ assert async_km_subclass .call_count ("is_alive" ) >= 1
551
567
552
568
is_alive = await async_km_subclass .is_alive ()
553
569
assert is_alive is False
570
+ assert async_km_subclass .call_count ("is_alive" ) >= 1
554
571
assert async_km_subclass .context .closed
0 commit comments