Fix(Postgres): Set INDEX_OFFSET to 1 (#1782)

* Add failing test for array offset * Add INDEX_OFFSET = 1 for Postgres
tobymao · Jun 15, 2023 · 4a1068b · 4a1068b
1 parent 3b95a46
commit 4a1068b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/sqlglot/dialects/postgres.py b/sqlglot/dialects/postgres.py
@@ -183,6 +183,7 @@ def _to_timestamp(args: t.List) -> exp.Expression:
 
 
 class Postgres(Dialect):
+    INDEX_OFFSET = 1
     NULL_ORDERING = "nulls_are_large"
     TIME_FORMAT = "'YYYY-MM-DD HH24:MI:SS'"
     TIME_MAPPING = {

diff --git a/tests/dialects/test_postgres.py b/tests/dialects/test_postgres.py
@@ -107,6 +107,16 @@ def test_unnest(self):
             },
         )
 
+    def test_array_offset(self):
+        self.validate_all(
+            "SELECT col[1]",
+            write={
+                "hive": "SELECT col[0]",
+                "postgres": "SELECT col[1]",
+                "presto": "SELECT col[1]",
+            },
+        )
+
     def test_postgres(self):
         self.validate_identity("CAST(x AS INT4RANGE)")
         self.validate_identity("CAST(x AS INT4MULTIRANGE)")