Fix for comaparison of AnyUrl objects (#11082)

alexprabhat99 · sydney-runkle · commit 9faa8d9cbd32 · 2024-12-18T10:30:03.000-05:00
diff --git a/pydantic/networks.py b/pydantic/networks.py
@@ -224,6 +224,18 @@ def __deepcopy__(self, memo: dict) -> Self:
     def __eq__(self, other: Any) -> bool:
         return self.__class__ is other.__class__ and self._url == other._url
 
+    def __lt__(self, other: Any) -> bool:
+        return self.__class__ is other.__class__ and self._url < other._url
+
+    def __gt__(self, other: Any) -> bool:
+        return self.__class__ is other.__class__ and self._url > other._url
+
+    def __le__(self, other: Any) -> bool:
+        return self.__class__ is other.__class__ and self._url <= other._url
+
+    def __ge__(self, other: Any) -> bool:
+        return self.__class__ is other.__class__ and self._url >= other._url
+
     def __hash__(self) -> int:
         return hash(self._url)
 
diff --git a/tests/test_networks.py b/tests/test_networks.py
@@ -1147,6 +1147,16 @@ def test_json_schema() -> None:
     assert ser_json_schema == {'type': 'string', 'format': 'uri', 'minLength': 1, 'maxLength': 2083}
 
 
+def test_any_url_comparison() -> None:
+    first_url = AnyUrl('https://a.com')
+    second_url = AnyUrl('https://b.com')
+
+    assert first_url < second_url
+    assert second_url > first_url
+    assert first_url <= second_url
+    assert second_url >= first_url
+
+
 def test_max_length_base_url() -> None:
     class Model(BaseModel):
         url: AnyUrl = Field(max_length=20)
