diff --git a/src/tests/test_command.py b/src/tests/test_command.py index 487725dc5..808f5c9f1 100644 --- a/src/tests/test_command.py +++ b/src/tests/test_command.py @@ -18,6 +18,7 @@ def sql_magic(ip): ("sqlite://", "", "", "sqlite://", None), ("SELECT * FROM TABLE", "", "SELECT * FROM TABLE\n", "", None), ("SELECT * FROM", "TABLE", "SELECT * FROM\nTABLE", "", None), + ("my_var << SELECT * FROM table", "", "SELECT * FROM table\n", "", "my_var"), ("my_var << SELECT *", "FROM table", "SELECT *\nFROM table", "", "my_var"), ("[db]", "", "", "sqlite://", None), ], @@ -26,7 +27,8 @@ def sql_magic(ip): "connection-string", "sql-query", "sql-query-in-line-and-cell", - "parsed-var", + "parsed-var-single-line", + "parsed-var-multi-line", "config", ], )