diff --git a/tests/vector_stores/test_weaviate.py b/tests/vector_stores/test_weaviate.py index d4ace1a1b..e7dcd3b6d 100644 --- a/tests/vector_stores/test_weaviate.py +++ b/tests/vector_stores/test_weaviate.py @@ -202,6 +202,14 @@ class TestWeaviateDB(unittest.TestCase): self.client_mock.collections.delete.assert_called_once_with("test_collection") + def test_reset(self): + self.client_mock.collections.exists.return_value = False + + self.weaviate_db.reset() + + self.client_mock.collections.delete.assert_called_once_with("test_collection") + self.client_mock.collections.create.assert_called_once() + if __name__ == "__main__": unittest.main()