diff options
author | mike bayer <mike_mp@zzzcomputing.com> | 2022-11-29 23:35:16 +0000 |
---|---|---|
committer | Gerrit Code Review <gerrit@ci3.zzzcomputing.com> | 2022-11-29 23:35:16 +0000 |
commit | 7857a1de32169858367446d11089c34f8daee957 (patch) | |
tree | 48a35acde7cfb298de8d52102f45a2afcdff312b /test/dialect/postgresql/test_compiler.py | |
parent | 78833af4e650d37e6257cfbb541e4db56e2a285f (diff) | |
parent | 0b239579f03c82f7669d77c238e4fda8638fb9c3 (diff) | |
download | sqlalchemy-7857a1de32169858367446d11089c34f8daee957.tar.gz |
Merge "Add value-level hooks for SQL type detection; apply to Range" into main
Diffstat (limited to 'test/dialect/postgresql/test_compiler.py')
-rw-r--r-- | test/dialect/postgresql/test_compiler.py | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/test/dialect/postgresql/test_compiler.py b/test/dialect/postgresql/test_compiler.py index 431cd7ded..ee3372c74 100644 --- a/test/dialect/postgresql/test_compiler.py +++ b/test/dialect/postgresql/test_compiler.py @@ -42,6 +42,7 @@ from sqlalchemy.dialects.postgresql import ExcludeConstraint from sqlalchemy.dialects.postgresql import insert from sqlalchemy.dialects.postgresql import JSONB from sqlalchemy.dialects.postgresql import JSONPATH +from sqlalchemy.dialects.postgresql import Range from sqlalchemy.dialects.postgresql import TSRANGE from sqlalchemy.dialects.postgresql.base import PGDialect from sqlalchemy.dialects.postgresql.psycopg2 import PGDialect_psycopg2 @@ -2397,6 +2398,26 @@ class CompileTest(fixtures.TestBase, AssertsCompiledSQL): "AS jsonb_path_exists_1 FROM data", ) + def test_custom_object_hook(self): + # See issue #8884 + from datetime import date + + usages = table( + "usages", + column("id", Integer), + column("date", Date), + column("amount", Integer), + ) + period = Range(date(2022, 1, 1), (2023, 1, 1)) + stmt = select(func.sum(usages.c.amount)).where( + usages.c.date.op("<@")(period) + ) + self.assert_compile( + stmt, + "SELECT sum(usages.amount) AS sum_1 FROM usages " + "WHERE usages.date <@ %(date_1)s::DATERANGE", + ) + class InsertOnConflictTest(fixtures.TablesTest, AssertsCompiledSQL): __dialect__ = postgresql.dialect() |