[AUR-430] Add test case for Chroma VectoStore save/load (#26)

* add test case for Chroma save/load

* minor name change

* add delete_collection support for chroma

* move save load to chroma

---------

Co-authored-by: Nguyen Trung Duc (john) <john@cinnamon.is>

tests/test_vectorstore.py

@@ -62,8 +62,31 @@ class TestChromaVectorStore:
         _, _, out_ids = db.query(embedding=[0.42, 0.52, 0.53], top_k=1)
         assert out_ids == ["b"]
 
+    def test_save_load_delete(self, tmp_path):
+        """Test that save/load func behave correctly."""
+        embeddings = [[0.1, 0.2, 0.3], [0.4, 0.5, 0.6], [0.7, 0.8, 0.9]]
+        metadatas = [{"a": 1, "b": 2}, {"a": 3, "b": 4}, {"a": 5, "b": 6}]
+        ids = ["1", "2", "3"]
+        db = ChromaVectorStore(path=str(tmp_path))
+        db.add(embeddings=embeddings, metadatas=metadatas, ids=ids)
+        db.save()
 
-class TestSimpleVectorStore:
+        db2 = ChromaVectorStore(path=str(tmp_path))
+        db2.load()
+        assert (
+            db2._collection.count() == 3
+        ), "load function does not load data completely"
+
+        # test delete collection function
+        db2.delete_collection()
+        # reinit the chroma with the same collection name
+        db2 = ChromaVectorStore(path=str(tmp_path))
+        assert (
+            db2._collection.count() == 0
+        ), "delete collection function does not work correctly"
+
+
+class TestInMemoryVectorStore:
     def test_add(self):
         """Test that add func adds correctly."""
 
@@ -88,7 +111,7 @@ class TestSimpleVectorStore:
         data = json.load(f)
         assert (
             "1" and "2" in data["text_id_to_ref_doc_id"]
-        ), "persist function does not save data completely"
+        ), "save function does not save data completely"
         assert (
             "3" not in data["text_id_to_ref_doc_id"]
         ), "delete function does not delete data completely"
@@ -98,4 +121,4 @@ class TestSimpleVectorStore:
             0.4,
             0.5,
             0.6,
-        ], "persist function does not load data completely"
+        ], "load function does not load data completely"