diff --git a/tests/test_upsert.py b/tests/test_upsert.py index 9b93f88..edf7f29 100644 --- a/tests/test_upsert.py +++ b/tests/test_upsert.py @@ -18,9 +18,15 @@ def test_upsert_all(fresh_db): def test_upsert_compound_primary_key(fresh_db): table = fresh_db["table"] - table.upsert_all([{"species": "dog", "id": 1, "name": "Cleo", "age": 4}, {"species": "cat", "id": 1, "name": "Catbag"}], pk=("species", "id")) + table.upsert_all( + [ + {"species": "dog", "id": 1, "name": "Cleo", "age": 4}, + {"species": "cat", "id": 1, "name": "Catbag"}, + ], + pk=("species", "id"), + ) table.upsert_all([{"species": "dog", "id": 1, "age": 5}], pk=("species", "id")) assert [ {"species": "dog", "id": 1, "name": "Cleo", "age": 5}, {"species": "cat", "id": 1, "name": "Catbag", "age": None}, - ] == list(table.rows) \ No newline at end of file + ] == list(table.rows)