From a293a37ef87182d7992fb701f5e05ef9d483195a Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Sun, 9 Feb 2025 23:29:21 +0000 Subject: [PATCH] test: Handle both directory removal and empty directory cases Co-Authored-By: Joe Moura --- tests/memory/test_memory_reset.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/memory/test_memory_reset.py b/tests/memory/test_memory_reset.py index a8d5dd3ea..6fc968caa 100644 --- a/tests/memory/test_memory_reset.py +++ b/tests/memory/test_memory_reset.py @@ -78,4 +78,5 @@ def test_memory_reset_cleans_up_files(temp_db_dir): memory.save("test memory", {"test": "metadata"}) assert any(temp_db_dir.iterdir()) # Directory should have files memory.reset() - assert not any(temp_db_dir.iterdir()) # Directory should be empty + # After reset, directory should either not exist or be empty + assert not os.path.exists(temp_db_dir) or not any(temp_db_dir.iterdir())