remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-09-27 20:34:12 +02:00
Code Issues Projects Releases Packages Wiki Activity

Refactored tests into new test_rows.py, refs #76

Browse source
This commit is contained in:
Simon Willison 2020-04-15 18:04:51 -07:00
commit fc38868bd4
3 changed files with 29 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

21
tests/test_get.py
View file

@ -29,24 +29,3 @@ def test_get_not_found(argument, expected_msg, fresh_db):
fresh_db["dogs"].get(argument) fresh_db["dogs"].get(argument)
if expected_msg is not None: if expected_msg is not None:
assert expected_msg == excinfo.value.args[0] assert expected_msg == excinfo.value.args[0]
@pytest.mark.parametrize(
"where,where_args,expected_ids",
[
("name = ?", ["Pancakes"], {2}),
("age > ?", [3], {1}),
("name is not null", [], {1, 2}),
("is_good = ?", [True], {1, 2}),
],
)
def test_rows_where(where, where_args, expected_ids, fresh_db):
table = fresh_db["dogs"]
table.insert_all(
[
{"id": 1, "name": "Cleo", "age": 4, "is_good": True},
{"id": 2, "name": "Pancakes", "age": 3, "is_good": True},
],
pk="id",
)
assert expected_ids == {r["id"] for r in table.rows_where(where, where_args)}

6
tests/test_introspect.py
View file

@ -62,12 +62,6 @@ def test_columns(existing_db):
] ]
def test_rows(existing_db):
assert [{"text": "one"}, {"text": "two"}, {"text": "three"}] == list(
existing_db["foo"].rows
)
def test_schema(existing_db): def test_schema(existing_db):
assert "CREATE TABLE foo (text TEXT)" == existing_db["foo"].schema assert "CREATE TABLE foo (text TEXT)" == existing_db["foo"].schema

29
tests/test_rows.py Normal file
View file

@ -0,0 +1,29 @@
from sqlite_utils.db import Index, View
import pytest
def test_rows(existing_db):
assert [{"text": "one"}, {"text": "two"}, {"text": "three"}] == list(
existing_db["foo"].rows
)
@pytest.mark.parametrize(
"where,where_args,expected_ids",
[
("name = ?", ["Pancakes"], {2}),
("age > ?", [3], {1}),
("name is not null", [], {1, 2}),
("is_good = ?", [True], {1, 2}),
],
)
def test_rows_where(where, where_args, expected_ids, fresh_db):
table = fresh_db["dogs"]
table.insert_all(
[
{"id": 1, "name": "Cleo", "age": 4, "is_good": True},
{"id": 2, "name": "Pancakes", "age": 3, "is_good": True},
],
pk="id",
)
assert expected_ids == {r["id"] for r in table.rows_where(where, where_args)}