comparsion for AnyUrl objects

pydantic · sydney-runkle · Dec 11, 2024 · Dec 10, 2024 · Dec 11, 2024 · Dec 11, 2024
commit 1f63030988c882da4edf8a77169851e83a60fed7
diff --git a/pydantic/networks.py b/pydantic/networks.py
@@ -225,7 +225,16 @@ 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 str(self._url) < str(other._url)
+        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
@@ -1145,7 +1145,12 @@ def test_json_schema() -> None:
     ser_json_schema = ta.json_schema(mode='serialization')
     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 second_url > first_url
+
+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