aboutsummaryrefslogtreecommitdiffstats
path: root/test/apps/pci_optimization/test_ml_model.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/apps/pci_optimization/test_ml_model.py')
-rw-r--r--test/apps/pci_optimization/test_ml_model.py2
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'] = []