@@ -21,6 +21,7 @@ import {
2121import { IDisposable } from '../../platform/common/types' ;
2222import * as notebookUpdater from '../../kernels/execution/notebookUpdater' ;
2323import { createMockedNotebookDocument } from '../../test/datascience/editor-integration/helpers' ;
24+ import { WrappedError } from '../../platform/errors/types' ;
2425
2526suite ( 'DeepnoteNotebookCommandListener' , ( ) => {
2627 let commandListener : DeepnoteNotebookCommandListener ;
@@ -1138,7 +1139,11 @@ suite('DeepnoteNotebookCommandListener', () => {
11381139 } ) ;
11391140
11401141 // Call the method and expect rejection
1141- await assert . isRejected ( commandListener . addChartBlock ( ) , Error , 'No active notebook editor found' ) ;
1142+ await assert . isRejected (
1143+ commandListener . addChartBlock ( ) ,
1144+ WrappedError ,
1145+ 'No active notebook editor found'
1146+ ) ;
11421147 } ) ;
11431148
11441149 test ( 'should throw error when chainWithPendingUpdates fails' , async ( ) => {
@@ -1154,7 +1159,7 @@ suite('DeepnoteNotebookCommandListener', () => {
11541159 sandbox . stub ( notebookUpdater , 'chainWithPendingUpdates' ) . resolves ( false ) ;
11551160
11561161 // Call the method and expect rejection
1157- await assert . isRejected ( commandListener . addChartBlock ( ) , Error , 'Failed to insert chart block' ) ;
1162+ await assert . isRejected ( commandListener . addChartBlock ( ) , WrappedError , 'Failed to insert chart block' ) ;
11581163 } ) ;
11591164 } ) ;
11601165 } ) ;
0 commit comments