Fix tests

Author: elad-c

tests_pytest/common_tests/unit_tests/core/mixed_precision/resource_utilization_tools/test_resource_utilization_calculator.py

@@ -472,7 +472,6 @@ def prepare_compute_cuts(self, graph_mock, fw_impl_mock, fw_info_mock, mocker):
         graph_mock.nodes = nodes
         # use the Graph original method (need to bind it to graph_mock instance)
         graph_mock.find_node_by_name = MethodType(Graph.find_node_by_name, graph_mock)
-        graph_mock.retrieve_preserved_quantization_node = _identity_func
 
         # we should not use total size, setting it to bad number
         cut_elems1 = MemoryElements(elements={ActivationMemoryTensor(mp_reuse.output_shape, 'mp_reuse', 0)}, total_size=-1)

tests_pytest/conftest.py

@@ -31,7 +31,7 @@ def minimal_tpc():
 @fixture
 def graph_mock():
     """ Basic Graph mock. """
-    return Mock(spec_set=Graph, nodes=[])
+    return Mock(spec_set=Graph, nodes=[], retrieve_preserved_quantization_node=lambda x: x)
 
 
 @fixture