fix tests

Author: irenab

tests_pytest/common_tests/unit_tests/core/mixed_precision/search_methods/test_linear_programming_solver.py

@@ -115,7 +115,7 @@ def test_all_ru_targets(self):
         self._run_test(sensitivity, ru, ru_constraints, {'n1': 0, 'n2': 1, 'n3': 1})
 
         # in addition, increase activation ru for one of the cuts of the current optimal candidate of the 3rd layer
-        ru_constraints[RUTarget.ACTIVATION][6, 2] += 0.1
+        ru[RUTarget.ACTIVATION][6, 2] += 0.1
         self._run_test(sensitivity, ru, ru_constraints, {'n1': 0, 'n2': 1, 'n3': 0})
 
         # in addition, increase total ru for one of the cuts of the optimal candidate of the 2nd layer
@@ -124,7 +124,7 @@ def test_all_ru_targets(self):
 
         # in addition, increase bops for the optimal candidate of 2nd layer above constraint
         ru[RUTarget.BOPS][4, 0] += 0.1
-        self._run_test(sensitivity, ru, ru_constraints, {'n1': 2, 'n2': 2, 'n3': 0})
+        self._run_test(sensitivity, ru, ru_constraints, {'n1': 2, 'n2': 0, 'n3': 0})
 
     def _run_test(self, sensitivity, ru, ru_constraints, exp_res):
         solver = MixedPrecisionIntegerLPSolver(sensitivity, ru, ru_constraints)

tests_pytest/common_tests/unit_tests/core/mixed_precision/test_greedy_solution_refinement.py

@@ -15,6 +15,8 @@
 from typing import Dict, List
 
 from model_compression_toolkit.core import ResourceUtilization
+from model_compression_toolkit.core.common.mixed_precision.mixed_precision_search_manager import \
+    MixedPrecisionSearchManager
 from model_compression_toolkit.core.common.mixed_precision.solution_refinement_procedure import greedy_solution_refinement_procedure
 
 import pytest
@@ -27,11 +29,8 @@ def search_manager():
     manager = Mock()
     manager.mp_topo_configurable_nodes = MagicMock()
     manager.fw_info.get_kernel_op_attributes = MagicMock()
-    manager.replace_config_in_index = MagicMock(
-        side_effect=lambda config, idx, candidate: (
-            lambda new_config: (new_config.__setitem__(idx, candidate), new_config)[1]
-        )(copy.deepcopy(config))
-    )
+    manager.copy_config_with_replacement = MixedPrecisionSearchManager.copy_config_with_replacement
+    manager.using_virtual_graph = False    # asserted in refinement
     return manager
 
 
@@ -99,18 +98,18 @@ def test_greedy_solution_refinement_procedure(
     weight_bits_dict_1 = {'kernel': 4}
     act_bits_1 = 8
 
-    initial_solution = [1]
-    expected_solution = [1]
-
     node_mock = Mock()
     node_mock.candidates_quantization_cfg = candidate_configs(weight_bits_dict_0, act_bits_0, weight_bits_dict_1, act_bits_1)
 
+    initial_solution = {node_mock: 1}
+    expected_solution = {node_mock: 1}
+
     search_manager.mp_topo_configurable_nodes = [node_mock]
 
     search_manager.compute_resource_utilization_for_config = MagicMock(side_effect=lambda config: {
         0: ResourceUtilization(**alternative_candidate_resources_usage),
         1: ResourceUtilization(weights_memory=50, activation_memory=50),
-    }[config[-1]])
+    }[config[node_mock]])
 
     target_resource_utilization = ResourceUtilization(**resource_limit)