Fix tests for --execute command

dbcli · Nov 5, 2016 · 00fa9b2 · 00fa9b2
1 parent dbe05e8
commit 00fa9b2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_main.py b/tests/test_main.py
@@ -53,7 +53,7 @@ def test_execute_arg(executor):
 
     expected = 'a\nabc\n'
 
-    assert result.output == expected
+    assert expected in result.output
 
 
 @dbtest
@@ -67,7 +67,7 @@ def test_execute_arg_with_table(executor):
     expected = '+-----+\n| a   |\n|-----|\n| abc |\n+-----+\n'
 
     assert result.exit_code == 0
-    assert result.output == expected
+    assert expected in result.output
 
 
 @dbtest
@@ -81,7 +81,7 @@ def test_execute_arg_with_csv(executor):
     expected = 'a\nabc\n\n'
 
     assert result.exit_code == 0
-    assert result.output == expected
+    assert expected in result.output
 
 
 @dbtest