Add \G as a temporary way to use expanded mode.

This is similar to mycli.

1 files changed, 7 insertions, 0 deletions

diff --git a/tests/test_pgexecute.py b/tests/test_pgexecute.py
index 55d7f548..c42bcee3 100644
--- a/tests/test_pgexecute.py
+++ b/tests/test_pgexecute.py
@@ -86,6 +86,13 @@ def test_bools_are_treated_as_strings(executor):
         SELECT 1"""
     )
 
+@dbtest
+def test_expanded_slash_G(executor, pgspecial):
+    # Tests whether we reset the expanded output after a \G.
+    run(executor, """create table test(a boolean)""")
+    run(executor, """insert into test values(True)""")
+    results = run(executor, """select * from test \G""", pgspecial=pgspecial)
+    assert pgspecial.expanded_output == False
 
 @dbtest
 def test_schemata_table_views_and_columns_query(executor):