fix: fix cache tests

.gitignore

@@ -25,4 +25,7 @@ pandasai.egg-info
 .idea
 
 # cache
-cache.db
+
+# cache
+cache.db
+cache_*.db

pandasai/helpers/cache.py

@@ -1,5 +1,6 @@
 """Cache module for caching queries."""
 
+import os
 import shelve
 
 
@@ -55,3 +56,9 @@ def clear(self) -> None:
         """Clean the cache."""
 
         self.cache.clear()
+
+    def destroy(self) -> None:
+        """Destroy the cache."""
+
+        self.cache.close()
+        os.remove(self.filename + ".db")

tests/helpers/test_cache.py

@@ -1,12 +1,17 @@
+import uuid
+
 from pandasai.helpers.cache import Cache
 
 
 class TestCache:
     def test_cache(self):
-        cache = Cache()
+        cache_filename = f"cache_{uuid.uuid4().hex}"
+        cache = Cache(filename=cache_filename)
         cache.set("key", "value")
         assert cache.get("key") == "value"
 
         cache.delete("key")
         print(cache.get("key"))
         assert cache.get("key") is None
+
+        cache.destroy()