Reformat POL to standardized format

[osm/POL.git] / osm_policy_module / tests / unit / utils / test_vnfd_utils.py

diff --git a/osm_policy_module/tests/unit/utils/test_vnfd_utils.py b/osm_policy_module/tests/unit/utils/test_vnfd_utils.py
index 049d293..1474a1c 100644 (file)
--- a/osm_policy_module/tests/unit/utils/test_vnfd_utils.py
+++ b/osm_policy_module/tests/unit/utils/test_vnfd_utils.py
@@ -31,16 +31,20 @@ from osm_policy_module.core.exceptions import ManagementVduNotFound
 
 class VnfdUtilsTest(unittest.TestCase):
     def test_get_mgmt_vdu_on_valid_descriptor(self):
-        example_file = os.path.join(os.path.dirname(__file__), 'examples/cirros_vdu_scaling_vnfd.yaml')
-        with open(example_file, 'r') as file:
-            vnfd = yaml.safe_load(file)['vnfd']
+        example_file = os.path.join(
+            os.path.dirname(__file__), "examples/cirros_vdu_scaling_vnfd.yaml"
+        )
+        with open(example_file, "r") as file:
+            vnfd = yaml.safe_load(file)["vnfd"]
             vdu = VnfdUtils.get_mgmt_vdu(vnfd)
-            self.assertEqual(vdu['id'], 'cirros_vnfd-VM')
+            self.assertEqual(vdu["id"], "cirros_vnfd-VM")
 
     def test_get_mgmt_vdu_on_invalid_descriptor(self):
-        example_file = os.path.join(os.path.dirname(__file__), 'examples/cirros_vdu_scaling_vnfd.yaml')
-        with open(example_file, 'r') as file:
-            vnfd = yaml.safe_load(file)['vnfd']
-            vnfd['mgmt-cp'] = 'invalid-cp'
+        example_file = os.path.join(
+            os.path.dirname(__file__), "examples/cirros_vdu_scaling_vnfd.yaml"
+        )
+        with open(example_file, "r") as file:
+            vnfd = yaml.safe_load(file)["vnfd"]
+            vnfd["mgmt-cp"] = "invalid-cp"
             with self.assertRaises(ManagementVduNotFound):
                 VnfdUtils.get_mgmt_vdu(vnfd)