From dbdc2d26632175ab01d345b88ebf02ee8f95454a Mon Sep 17 00:00:00 2001 From: Niranjana Date: Mon, 16 Nov 2020 17:19:01 +0530 Subject: Fix issue in pci-ml model - Remove list conversion from fixed cells Issue-ID: OPTFRA-871 Signed-off-by: Niranjana Change-Id: If29945fe7513ea0fdb84bcabbdf5d546bfd8ac55 --- test/apps/pci_optimization/test_ml_model.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'test/apps/pci_optimization/test_ml_model.py') diff --git a/test/apps/pci_optimization/test_ml_model.py b/test/apps/pci_optimization/test_ml_model.py index 9c617a9..3a7f76c 100644 --- a/test/apps/pci_optimization/test_ml_model.py +++ b/test/apps/pci_optimization/test_ml_model.py @@ -48,7 +48,7 @@ class TestMlModel(unittest.TestCase): 'NEIGHBORS': [], 'NUM_SECOND_LEVEL_NEIGHBORS': 1, 'SECOND_LEVEL_NEIGHBORS': [], - 'PCI_UNCHANGEABLE_CELLS': [], + 'PCI_UNCHANGEABLE_CELLS': {}, 'ORIGINAL_PCIS': [] } @@ -70,18 +70,18 @@ class TestMlModel(unittest.TestCase): self.Mock_req = self.patcher_req.start() mlmodel = MlModel() mlmodel.get_additional_inputs(dzn_data, network_cell_info) - self.assertEqual(['Chn0001'], dzn_data['PCI_UNCHANGEABLE_CELLS']) + self.assertEqual({'Chn0001'}, dzn_data['PCI_UNCHANGEABLE_CELLS']) self.patcher_req.stop() dzn_data['PCI_UNCHANGEABLE_CELLS'] = [] self.patcher_req = patch('osdf.adapters.dcae.des.extract_data', side_effect=DESException('error')) self.Mock_req = self.patcher_req.start() mlmodel.get_additional_inputs(dzn_data, network_cell_info) - self.assertEqual([], dzn_data['PCI_UNCHANGEABLE_CELLS']) + self.assertEqual(set() , dzn_data['PCI_UNCHANGEABLE_CELLS']) self.patcher_req.stop() self.patcher_req = patch('osdf.adapters.dcae.des.extract_data', return_value=[]) self.Mock_req = self.patcher_req.start() mlmodel.get_additional_inputs(dzn_data, network_cell_info) - self.assertEqual([], dzn_data['PCI_UNCHANGEABLE_CELLS']) + self.assertEqual(set() , dzn_data['PCI_UNCHANGEABLE_CELLS']) self.patcher_req.stop() -- cgit 1.2.3-korg