testing column names are visible. closes #50

ploomber · Dec 31, 2022 · 0ea506f · 0ea506f
1 parent b51d9e3
commit 0ea506f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/tests/conftest.py b/src/tests/conftest.py
@@ -55,6 +55,7 @@ def ip():
             "CREATE TABLE author (first_name, last_name, year_of_death)",
             "INSERT INTO author VALUES ('William', 'Shakespeare', 1616)",
             "INSERT INTO author VALUES ('Bertold', 'Brecht', 1956)",
+            "CREATE TABLE empty_table (column INT, another INT)",
         ],
     )
     yield ip_session

diff --git a/src/tests/test_magic.py b/src/tests/test_magic.py
@@ -391,6 +391,13 @@ def test_close_connection(ip, tmp_empty):
     assert "sqlite:///two.db" not in Connection.connections
 
 
+def test_column_names_visible(ip, tmp_empty):
+    res = ip.run_line_magic("sql", "SELECT * FROM empty_table")
+
+    assert "<th>column</th>" in res._repr_html_()
+    assert "<th>another</th>" in res._repr_html_()
+
+
 @pytest.mark.xfail(reason="known parse @ parser.py error")
 def test_sqlite_path_with_spaces(ip, tmp_empty):
     ip.run_cell("%sql sqlite:///some database.db")