databricks bug fixes (#3416)

2025-09-05 05:52:09 -05:00
parent b27879bfd4
commit 9477184582
2 changed files with 5 additions and 4 deletions
@@ -300,8 +300,9 @@ def test_list_memories(db_instance_delta, mock_workspace_client):
        result=SimpleNamespace(data_array=[row])
    )
    res = db_instance_delta.list(limit=1)
-    assert len(res) == 1
-    assert res[0].id == "id3"
+    assert isinstance(res, list)
+    assert len(res[0]) == 1
+    assert res[0][0].id == "id3"


 # ---------------------- Reset Tests ---------------------- #