fix imports

cisco-en-programmability · Dec 13, 2024 · 597a8a6 · 597a8a6
1 parent ea0a18a
commit 597a8a6
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/catalystwan/integration_tests/profile_builder/test_pb_uc_voice.py b/catalystwan/integration_tests/profile_builder/test_pb_uc_voice.py
@@ -7,8 +7,12 @@
     TranslationProfileParcel,
     TranslationRuleParcel,
 )
+from catalystwan.models.configuration.feature_profile.sdwan.uc_voice.analog_interface import (
+    Association,
+    ModuleType,
+    SlotId,
+)
 from catalystwan.models.configuration.feature_profile.sdwan.uc_voice.translation_rule import Action, RuleSettings
-from catalystwan.models.configuration.feature_profile.sdwan.uc_voice.analog_interface import Association, SlotId, ModuleType
 from catalystwan.tests.builders.uc_voice import as_default
 
 
@@ -21,7 +25,9 @@ def setUp(self) -> None:
             feature_profile=FeatureProfileCreationPayload(name=self.fp_name, description=self.fp_description)
         )
         self.api = self.session.api.sdwan_feature_profiles.transport
-        self.tp = TranslationProfileParcel(parcel_name="TPP", parcel_description="TTP_Desc", translation_profile_settings=[])
+        self.tp = TranslationProfileParcel(
+            parcel_name="TPP", parcel_description="TTP_Desc", translation_profile_settings=[]
+        )
         self.tr_calling = TranslationRuleParcel(
             parcel_name="2",
             parcel_description="desc",
@@ -57,7 +63,9 @@ def test_when_build_profile_with_translation_profile_and_rules_expect_success(se
         # Assert
         assert len(report.failed_parcels) == 0
 
-    def test_when_build_profile_with_analog_interface_and_translation_profile_and_rules_assosiations_expect_success(self):
+    def test_when_build_profile_with_analog_interface_and_translation_profile_and_rules_assosiations_expect_success(
+        self,
+    ):
         # Arrange
         ai = AnalogInterfaceParcel(
             parcel_name="Ai",
@@ -71,9 +79,9 @@ def test_when_build_profile_with_analog_interface_and_translation_profile_and_ru
                     translation_profile=RefIdItem(ref_id=as_global("TPP")),
                     trunk_group=RefIdItem(ref_id=as_default(None)),
                     trunk_group_priority=as_default(None),
-                    translation_rule_direction=as_default(None)
+                    translation_rule_direction=as_default(None),
                 )
-            ]
+            ],
         )
         self.builder.add_translation_profile(self.tp, self.tr_calling, self.tr_called)
         self.builder.add_parcel_with_associations(ai)