Fix some failing tests.

1 files changed, 4 insertions, 4 deletions

diff --git a/tests/test_main.py b/tests/test_main.py
index af0b880..d8d01ed 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -27,14 +27,14 @@ def test_execute_arg(executor):
     result = runner.invoke(cli, args=CLI_ARGS + ["-e", sql])
 
     assert result.exit_code == 0
-    assert "a\nabc\n" in result.output
+    assert '"abc"' in result.output
 
     result = runner.invoke(cli, args=CLI_ARGS + ["--execute", sql])
 
     assert result.exit_code == 0
-    assert "a\nabc\n" in result.output
+    assert '"abc"' in result.output
 
-    expected = "a\nabc\n"
+    expected = '"a"\n"abc"\n'
 
     assert expected in result.output
 
@@ -78,7 +78,7 @@ def test_batch_mode(executor):
     result = runner.invoke(cli, args=CLI_ARGS, input=sql)
 
     assert result.exit_code == 0
-    assert "count(*)\n3\na\nabc\n" in "".join(result.output)
+    assert '"count(*)"\n"3"\n"a"\n"abc"\n' in "".join(result.output)
 
 
 @dbtest