Enable black in tox.ini

[osm/PLA.git] / osm_pla / test / test_mznPlacementConductor.py

diff --git a/osm_pla/test/test_mznPlacementConductor.py b/osm_pla/test/test_mznPlacementConductor.py
index 6bc42b6..9a181b1 100644 (file)
--- a/osm_pla/test/test_mznPlacementConductor.py
+++ b/osm_pla/test/test_mznPlacementConductor.py
@@ -13,6 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import logging
+
 # from collections import Counter
 from unittest import TestCase, mock
 
@@ -172,47 +173,89 @@ solve satisfy;
 
 class TestMznPlacementConductor(TestCase):
     def test__run_placement_model(self):
-        expected_result = [{'vimAccountId': 'bbbbbbbb-38f5-438d-b8ee-3f93b3531f87', 'member-vnf-index': '1'},
-                           {'vimAccountId': 'aaaaaaaa-38f5-438d-b8ee-3f93b3531f87', 'member-vnf-index': '2'},
-                           {'vimAccountId': 'aaaaaaaa-38f5-438d-b8ee-3f93b3531f87', 'member-vnf-index': '3'},
-                           {'vimAccountId': 'aaaaaaaa-38f5-438d-b8ee-3f93b3531f87', 'member-vnf-index': '4'}]
+        expected_result = [
+            {
+                "vimAccountId": "bbbbbbbb-38f5-438d-b8ee-3f93b3531f87",
+                "member-vnf-index": "1",
+            },
+            {
+                "vimAccountId": "aaaaaaaa-38f5-438d-b8ee-3f93b3531f87",
+                "member-vnf-index": "2",
+            },
+            {
+                "vimAccountId": "aaaaaaaa-38f5-438d-b8ee-3f93b3531f87",
+                "member-vnf-index": "3",
+            },
+            {
+                "vimAccountId": "aaaaaaaa-38f5-438d-b8ee-3f93b3531f87",
+                "member-vnf-index": "4",
+            },
+        ]
 
         mpc = MznPlacementConductor(logging.getLogger(__name__))
         placement = mpc._run_placement_model(mzn_model=test_mzn_model, ns_desc={})
         # sort the result to ease assert with expected result
-        sorted_placement = sorted(placement, key=lambda k: k['member-vnf-index'])
-        self.assertEqual(expected_result, sorted_placement, 'Faulty syntax or content')
+        sorted_placement = sorted(placement, key=lambda k: k["member-vnf-index"])
+        self.assertEqual(expected_result, sorted_placement, "Faulty syntax or content")
 
     def test__run_placement_model_w_pinning(self):
-        expected_result = [{'vimAccountId': 'bbbbbbbb-38f5-438d-b8ee-3f93b3531f87', 'member-vnf-index': '1'},
-                           {'vimAccountId': 'bbbbbbbb-38f5-438d-b8ee-3f93b3531f87', 'member-vnf-index': '2'},
-                           {'vimAccountId': 'bbbbbbbb-38f5-438d-b8ee-3f93b3531f87', 'member-vnf-index': '3'},
-                           {'vimAccountId': 'aaaaaaaa-38f5-438d-b8ee-3f93b3531f87', 'member-vnf-index': '4'}]
-
-        ns_desc = [{'vnf_price_per_vim': [10, 9, 7, 8], 'vnf_id': '2'},
-                   {'vim_account': 'vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87', 'vnf_price_per_vim': [10, 9, 7, 8],
-                    'vnf_id': '1'},
-                   {'vnf_price_per_vim': [10, 9, 7, 8], 'vnf_id': '4'},
-                   {'vim_account': 'vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87', 'vnf_price_per_vim': [10, 9, 7, 8],
-                    'vnf_id': '3'}
-                   ]
+        expected_result = [
+            {
+                "vimAccountId": "bbbbbbbb-38f5-438d-b8ee-3f93b3531f87",
+                "member-vnf-index": "1",
+            },
+            {
+                "vimAccountId": "bbbbbbbb-38f5-438d-b8ee-3f93b3531f87",
+                "member-vnf-index": "2",
+            },
+            {
+                "vimAccountId": "bbbbbbbb-38f5-438d-b8ee-3f93b3531f87",
+                "member-vnf-index": "3",
+            },
+            {
+                "vimAccountId": "aaaaaaaa-38f5-438d-b8ee-3f93b3531f87",
+                "member-vnf-index": "4",
+            },
+        ]
+
+        ns_desc = [
+            {"vnf_price_per_vim": [10, 9, 7, 8], "vnf_id": "2"},
+            {
+                "vim_account": "vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87",
+                "vnf_price_per_vim": [10, 9, 7, 8],
+                "vnf_id": "1",
+            },
+            {"vnf_price_per_vim": [10, 9, 7, 8], "vnf_id": "4"},
+            {
+                "vim_account": "vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87",
+                "vnf_price_per_vim": [10, 9, 7, 8],
+                "vnf_id": "3",
+            },
+        ]
 
         mpc = MznPlacementConductor(logging.getLogger(__name__))
-        placement = mpc._run_placement_model(mzn_model=test_mzn_model_w_pinning, ns_desc=ns_desc)
+        placement = mpc._run_placement_model(
+            mzn_model=test_mzn_model_w_pinning, ns_desc=ns_desc
+        )
         # sort the result to ease assert with expected result
-        sorted_placement = sorted(placement, key=lambda k: k['member-vnf-index'])
-        self.assertEqual(expected_result, sorted_placement, 'Faulty syntax or content')
+        sorted_placement = sorted(placement, key=lambda k: k["member-vnf-index"])
+        self.assertEqual(expected_result, sorted_placement, "Faulty syntax or content")
 
     def test__run_placement_model_unsatisfiable(self):
         mpc = MznPlacementConductor(logging.getLogger(__name__))
-        self.assertEqual([{}], mpc._run_placement_model(mzn_model=test_mzn_unsatisfiable_model, ns_desc={}),
-                         "Faulty syntax or content for unsatisfiable model")
-
-    @mock.patch.object(MznModelGenerator, 'create_model', side_effect=['%model'])
-    @mock.patch.object(MznPlacementConductor, '_run_placement_model')
+        self.assertEqual(
+            [{}],
+            mpc._run_placement_model(
+                mzn_model=test_mzn_unsatisfiable_model, ns_desc={}
+            ),
+            "Faulty syntax or content for unsatisfiable model",
+        )
+
+    @mock.patch.object(MznModelGenerator, "create_model", side_effect=["%model"])
+    @mock.patch.object(MznPlacementConductor, "_run_placement_model")
     def test_do_placement_computation(self, mock_run, mock_create):
         mpc = MznPlacementConductor(logging.getLogger(__name__))
-        dummy_nspd = {'ns_desc': {}}
+        dummy_nspd = {"ns_desc": {}}
         _ = mpc.do_placement_computation(dummy_nspd)
         mock_create.assert_called_with(dummy_nspd)
-        mock_run.assert_called_with('%model', {})
+        mock_run.assert_called_with("%model", {})