Refactored tests into new test_rows.py, refs #76

tests/test_get.py

@@ -29,24 +29,3 @@ def test_get_not_found(argument, expected_msg, fresh_db):
         fresh_db["dogs"].get(argument)
     if expected_msg is not None:
         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)}

tests/test_introspect.py

@@ -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):
     assert "CREATE TABLE foo (text TEXT)" == existing_db["foo"].schema
 

tests/test_rows.py

@@ -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)}