Skip to content

Commit 280332d

Browse files
committed
Fix cleanup code
1 parent d6c478a commit 280332d

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

tests/integration/test_spanner_loader.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def client() -> Client:
3535
return Client(project=project_id)
3636

3737

38-
@pytest.fixture()
38+
@pytest.fixture(scope="class")
3939
def cleanupGSQL(client):
4040
yield
4141

@@ -53,7 +53,7 @@ def cleanupGSQL(client):
5353
print("\nGSQL Cleanup complete.")
5454

5555

56-
@pytest.fixture()
56+
@pytest.fixture(scope="class")
5757
def cleanupPGSQL(client):
5858
yield
5959

tests/integration/test_spanner_vector_store.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -556,7 +556,7 @@ def test_delete(self, setup_database):
556556
)
557557

558558
docs = loader.load()
559-
deleted = db.delete(documents=[docs[0], docs[1]])
559+
deleted = db.delete(documents=docs)
560560

561561
assert deleted
562562

@@ -760,7 +760,7 @@ def test_spanner_vector_delete_data(self, setup_database):
760760

761761
docs = loader.load()
762762

763-
deleted = db.delete(documents=[docs[0], docs[1]])
763+
deleted = db.delete(documents=docs)
764764

765765
assert deleted
766766

0 commit comments

Comments
 (0)