Refactor tests to prevent Kgb warning

Author: roman-janik-nxp

backends/nxp/tests/ir/converter/node_converter/test_addmm_converter.py

@@ -30,7 +30,9 @@ def setUpClass(cls):
 
     def test_addmm_conversion(self):
         with kgb.spy_on(
-            EdgeProgramToIRConverter.convert_program, call_original=True
+            EdgeProgramToIRConverter.convert_program,
+            call_original=True,
+            owner=EdgeProgramToIRConverter,
         ) as converter_spy:
             input_shape = (1, 32)
             model = AddmmModule(input_shape[1])
@@ -60,7 +62,9 @@ def test_addmm_conversion(self):
 
     def test_linear_conversion__with_bias(self):
         with kgb.spy_on(
-            EdgeProgramToIRConverter.convert_program, call_original=True
+            EdgeProgramToIRConverter.convert_program,
+            call_original=True,
+            owner=EdgeProgramToIRConverter,
         ) as converter_spy:
             input_shape = (10, 32)
             model = LinearModule(bias=True)

backends/nxp/tests/ir/converter/node_converter/test_clone_converter.py

@@ -108,7 +108,9 @@ def test_conv_dropout_quant(self, inplace_dropout: bool, input_shape: tuple[int]
         ).eval()
 
         with kgb.spy_on(
-            EdgeProgramToIRConverter.convert_program, call_original=True
+            EdgeProgramToIRConverter.convert_program,
+            call_original=True,
+            owner=EdgeProgramToIRConverter,
         ) as converter_spy:
             quantized_program = to_quantized_edge_program(
                 model, input_shape
@@ -159,7 +161,9 @@ def test_clone_pool_view_copy_quant(self, input_shape: tuple[int] = (1, 64, 25,
         model = KWSFinalBlock(input_shape).eval()
 
         with kgb.spy_on(
-            EdgeProgramToIRConverter.convert_program, call_original=True
+            EdgeProgramToIRConverter.convert_program,
+            call_original=True,
+            owner=EdgeProgramToIRConverter,
         ) as converter_spy:
             quantized_program = to_quantized_edge_program(
                 model, input_shape

backends/nxp/tests/ir/converter/node_converter/test_mm_converter.py

@@ -30,7 +30,9 @@ def setUpClass(cls):
 
     def test_mm_conversion(self):
         with kgb.spy_on(
-            EdgeProgramToIRConverter.convert_program, call_original=True
+            EdgeProgramToIRConverter.convert_program,
+            call_original=True,
+            owner=EdgeProgramToIRConverter,
         ) as converter_spy:
             input_shape = (1, 32)
             model = MmModule(input_shape[1])
@@ -60,7 +62,9 @@ def test_mm_conversion(self):
 
     def test_linear_conversion__without_bias(self):
         with kgb.spy_on(
-            EdgeProgramToIRConverter.convert_program, call_original=True
+            EdgeProgramToIRConverter.convert_program,
+            call_original=True,
+            owner=EdgeProgramToIRConverter,
         ) as converter_spy:
             input_shape = (10, 32)
             model = LinearModule(bias=False)

backends/nxp/tests/ir/converter/node_converter/test_tanh_converter.py

@@ -48,7 +48,9 @@ def test_conv_tanh(
         self, _: str, inplace: bool, input_shape: tuple[int] = (1, 3, 112, 112)
     ):
         with kgb.spy_on(
-            EdgeProgramToIRConverter.convert_program, call_original=True
+            EdgeProgramToIRConverter.convert_program,
+            call_original=True,
+            owner=EdgeProgramToIRConverter,
         ) as converter_spy:
             if inplace:
                 model = Conv2dWithActivation(

backends/nxp/tests/test_per_channel_conversion.py

@@ -110,7 +110,9 @@ def setUpClass(cls):
 
     def test_per_channel_convolution(self):
         with kgb.spy_on(
-            EdgeProgramToIRConverter.convert_program, call_original=True
+            EdgeProgramToIRConverter.convert_program,
+            call_original=True,
+            owner=EdgeProgramToIRConverter,
         ) as converter_spy:
             model = Conv2dModule(
                 in_channels=8, out_channels=32, kernel_size=5, padding=3