test: move date-related tests to the date tests class

diff --git a/tests/types/test_datetime.py b/tests/types/test_datetime.py
index 11f421a38f4ee94e188aa76f3180a67e3366e3cd..6b5d7daf1dd8460e1abc07ad3f245d38a2e578b7 100644 (file)
--- a/tests/types/test_datetime.py
+++ b/tests/types/test_datetime.py
@@ -80,6 +80,37 @@ class TestDate:
         with pytest.raises(DataError):
             cur.fetchone()[0]
 
+    def test_infinity_date_example(self, conn):
+        # NOTE: this is an example in the docs. Make sure it doesn't regress when
+        # adding binary datetime adapters
+        from datetime import date
+        from psycopg.types.datetime import DateLoader, DateDumper
+
+        class InfDateDumper(DateDumper):
+            def dump(self, obj):
+                if obj == date.max:
+                    return b"infinity"
+                else:
+                    return super().dump(obj)
+
+        class InfDateLoader(DateLoader):
+            def load(self, data):
+                if data == b"infinity":
+                    return date.max
+                else:
+                    return super().load(data)
+
+        cur = conn.cursor()
+        cur.adapters.register_dumper(date, InfDateDumper)
+        cur.adapters.register_loader("date", InfDateLoader)
+
+        rec = cur.execute(
+            "SELECT %s::text, %s::text", [date(2020, 12, 31), date.max]
+        ).fetchone()
+        assert rec == ("2020-12-31", "infinity")
+        rec = cur.execute("select '2020-12-31'::date, 'infinity'::date").fetchone()
+        assert rec == (date(2020, 12, 31), date(9999, 12, 31))
+
 
 class TestDatetime:
     @pytest.mark.parametrize(
@@ -581,37 +612,6 @@ class TestInterval:
         with pytest.raises(DataError):
             cur.fetchone()[0]
 
-    def test_infinity_date_example(self, conn):
-        # NOTE: this is an example in the docs. Make sure it doesn't regress when
-        # adding binary datetime adapters
-        from datetime import date
-        from psycopg.types.datetime import DateLoader, DateDumper
-
-        class InfDateDumper(DateDumper):
-            def dump(self, obj):
-                if obj == date.max:
-                    return b"infinity"
-                else:
-                    return super().dump(obj)
-
-        class InfDateLoader(DateLoader):
-            def load(self, data):
-                if data == b"infinity":
-                    return date.max
-                else:
-                    return super().load(data)
-
-        cur = conn.cursor()
-        cur.adapters.register_dumper(date, InfDateDumper)
-        cur.adapters.register_loader("date", InfDateLoader)
-
-        rec = cur.execute(
-            "SELECT %s::text, %s::text", [date(2020, 12, 31), date.max]
-        ).fetchone()
-        assert rec == ("2020-12-31", "infinity")
-        rec = cur.execute("select '2020-12-31'::date, 'infinity'::date").fetchone()
-        assert rec == (date(2020, 12, 31), date(9999, 12, 31))
-
     def test_load_copy(self, conn):
         cur = conn.cursor(binary=False)
         with cur.copy(