Update test

chroma-core · Jan 9, 2025 · 654c768 · 654c768
1 parent 8de35f5
commit 654c768
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/chromadb/test/api/test_delete_database.py b/chromadb/test/api/test_delete_database.py
@@ -15,13 +15,19 @@ def test_deletes_database(client_factories: ClientFactories) -> None:
     admin_client.create_database("test_delete_database")
 
     client = client_factories.create_client(database="test_delete_database")
-    client.create_collection("foo")
+    collection = client.create_collection("foo")
 
     admin_client.delete_database("test_delete_database")
 
+    with pytest.raises(NotFoundError):
+        admin_client.get_database("test_delete_database")
+
     with pytest.raises(InvalidCollectionException):
         client.get_collection("foo")
 
+    with pytest.raises(InvalidCollectionException):
+        collection.upsert(["foo"], [0.0, 0.0, 0.0])
+
 
 def test_does_not_affect_other_databases(client_factories: ClientFactories) -> None:
     if not NOT_CLUSTER_ONLY: