testing << operator - closes #44

ploomber · Dec 31, 2022 · c666007 · c666007
1 parent 0ea506f
commit c666007
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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",
     ],
 )