.table_names and .tables properties plus expanded docs

This commit is contained in:
Simon Willison 2018-07-31 17:35:36 -07:00
commit 515d362ad6
5 changed files with 54 additions and 11 deletions

View file

@ -4,12 +4,12 @@ import json
def test_create_table(fresh_db):
assert [] == fresh_db.tables
assert [] == fresh_db.table_names
table = fresh_db.create_table(
"test_table",
{"text_col": str, "float_col": float, "int_col": int, "bool_col": bool},
)
assert ["test_table"] == fresh_db.tables
assert ["test_table"] == fresh_db.table_names
assert [
{"name": "text_col", "type": "TEXT"},
{"name": "float_col", "type": "FLOAT"},
@ -29,7 +29,7 @@ def test_create_table(fresh_db):
)
def test_create_table_from_example(fresh_db, example, expected_columns):
fresh_db["people"].insert(example)
assert ["people"] == fresh_db.tables
assert ["people"] == fresh_db.table_names
assert expected_columns == [
{"name": col.name, "type": col.type} for col in fresh_db["people"].columns
]