Fixed Test coverage issue

aristanetworks · Feb 10, 2025 · c4fd840 · c4fd840
1 parent e5f9f2c
commit c4fd840
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/anta/input_models/routing/bgp.py b/anta/input_models/routing/bgp.py
@@ -325,13 +325,5 @@ def __str__(self) -> str:
         --------
         - AFI-SAFI: IPv4 Unicast
         """
-        base_string = "AFI-SAFI:"
-        if self.afi_safi == "v4u":
-            base_string = f"{base_string} IPv4 Unicast"
-        elif self.afi_safi == "v4m":
-            base_string = f"{base_string} IPv4 Multicast"
-        elif self.afi_safi == "v6u":
-            base_string = f"{base_string} IPv6 Unicast"
-        elif self.afi_safi == "v6m":
-            base_string = f"{base_string} IPv6 Multicast"
-        return base_string
+        mappings = {"v4u": "IPv4 Unicast", "v4m": "IPv4 Multicast", "v6u": "IPv6 Unicast", "v6m": "IPv6 Multicast"}
+        return f"AFI-SAFI: {mappings[self.afi_safi]}"