diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/apps/pci_optimization/test_ml_model.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/test/apps/pci_optimization/test_ml_model.py b/test/apps/pci_optimization/test_ml_model.py index 3a7f76c..cf0dee5 100644 --- a/test/apps/pci_optimization/test_ml_model.py +++ b/test/apps/pci_optimization/test_ml_model.py @@ -70,7 +70,7 @@ 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({1}, dzn_data['PCI_UNCHANGEABLE_CELLS']) self.patcher_req.stop() dzn_data['PCI_UNCHANGEABLE_CELLS'] = [] |