Made tests more robust

googleapis · Sep 18, 2024 · 58c14da · 58c14da
1 parent 6a9f8dc
commit 58c14da
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_pg_to_alloy_migrator.py b/tests/test_pg_to_alloy_migrator.py
@@ -263,6 +263,7 @@ async def test_amigrate_pgvector_collection_error(
  )
  await aexecute(migrator, f"TRUNCATE TABLE {EMBEDDING_TABLE}")
  await aexecute(migrator, f"TRUNCATE TABLE {COLLECTIONS_TABLE}")
+ await aexecute(migrator, f"DROP TABLE IF EXISTS collection_1")
 
  async def test_amigrate_pgvector_collection_json_metadata(
  self, engine, migrator, sample_embeddings
@@ -575,6 +576,7 @@ async def test_migrate_pgvector_collection_error(self, migrator, sample_embeddin
  )
  await aexecute(migrator, f"TRUNCATE TABLE {EMBEDDING_TABLE}")
  await aexecute(migrator, f"TRUNCATE TABLE {COLLECTIONS_TABLE}")
+ await aexecute(migrator, "DROP TABLE IF EXISTS collection_1")
 
  async def test_migrate_pgvector_collection_json_metadata(
  self, engine, migrator, sample_embeddings