From 5f0517fb11f186582915d1d63878d11aa202168b Mon Sep 17 00:00:00 2001 From: Simon Willison Date: Sun, 25 Jun 2023 14:46:24 -0700 Subject: [PATCH] Updated tests to handle unexpected sqlite_stat4 table, refs #559 --- tests/test_analyze.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/tests/test_analyze.py b/tests/test_analyze.py index a47c8be..a4cd8a2 100644 --- a/tests/test_analyze.py +++ b/tests/test_analyze.py @@ -14,7 +14,9 @@ def db(fresh_db): def test_analyze_whole_database(db): assert set(db.table_names()) == {"one_index", "two_indexes"} db.analyze() - assert set(db.table_names()) == {"one_index", "two_indexes", "sqlite_stat1"} + assert set(db.table_names()).issuperset( + {"one_index", "two_indexes", "sqlite_stat1"} + ) assert list(db["sqlite_stat1"].rows) == [ {"tbl": "two_indexes", "idx": "idx_two_indexes_species", "stat": "1 1"}, {"tbl": "two_indexes", "idx": "idx_two_indexes_name", "stat": "1 1"}, @@ -24,13 +26,15 @@ def test_analyze_whole_database(db): @pytest.mark.parametrize("method", ("db_method_with_name", "table_method")) def test_analyze_one_table(db, method): - assert set(db.table_names()) == {"one_index", "two_indexes"} + assert set(db.table_names()).issuperset({"one_index", "two_indexes"}) if method == "db_method_with_name": db.analyze("one_index") elif method == "table_method": db["one_index"].analyze() - assert set(db.table_names()) == {"one_index", "two_indexes", "sqlite_stat1"} + assert set(db.table_names()).issuperset( + {"one_index", "two_indexes", "sqlite_stat1"} + ) assert list(db["sqlite_stat1"].rows) == [ {"tbl": "one_index", "idx": "idx_one_index_name", "stat": "1 1"} ] @@ -39,7 +43,9 @@ def test_analyze_one_table(db, method): def test_analyze_index_by_name(db): assert set(db.table_names()) == {"one_index", "two_indexes"} db.analyze("idx_two_indexes_species") - assert set(db.table_names()) == {"one_index", "two_indexes", "sqlite_stat1"} + assert set(db.table_names()).issuperset( + {"one_index", "two_indexes", "sqlite_stat1"} + ) assert list(db["sqlite_stat1"].rows) == [ {"tbl": "two_indexes", "idx": "idx_two_indexes_species", "stat": "1 1"}, ]