Update unit tests for VerifyBGPPeerCount

aristanetworks · Oct 25, 2024 · 01b26a0 · 01b26a0
1 parent c26fb82
commit 01b26a0
Show file tree

Hide file tree

Showing 2 changed files with 141 additions and 412 deletions.
diff --git a/anta/tests/routing/bgp_input_models.py b/anta/tests/routing/bgp_input_models.py
@@ -88,10 +88,17 @@ def validate_inputs(self) -> Self:
     @property
     def eos_key(self) -> str:
         """AFI/SAFI EOS key representation."""
+        # Pydantic handles the validation of the AFI/SAFI combination, so we can ignore error handling here.
         return AFI_SAFI_EOS_KEY[(self.afi, self.safi)]
 
     def __str__(self) -> str:
-        """Return a string representation of the BgpAddressFamily model. Used in failure messages."""
+        """Return a string representation of the BgpAddressFamily model. Used in failure messages.
+
+        Examples
+        --------
+        - AFI:ipv4 SAFI:unicast VRF:default
+        - AFI:evpn
+        """
         base_string = f"AFI:{self.afi}"
         if self.safi is not None:
             base_string += f" SAFI:{self.safi}"