From 39f9c8ce57b019cbc4401b5a689d1aa4c719aac0 Mon Sep 17 00:00:00 2001 From: Eric Pinzur Date: Thu, 10 Oct 2024 15:25:37 +0200 Subject: [PATCH] fix tests --- libs/astradb/tests/integration_tests/test_graphvectorstore.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/astradb/tests/integration_tests/test_graphvectorstore.py b/libs/astradb/tests/integration_tests/test_graphvectorstore.py index 60bd4e1..544ffd0 100644 --- a/libs/astradb/tests/integration_tests/test_graphvectorstore.py +++ b/libs/astradb/tests/integration_tests/test_graphvectorstore.py @@ -353,9 +353,9 @@ def test_gvs_add_nodes( assert hits[0].page_content == "[0, 2]" md0 = hits[0].metadata assert md0["m"] == 0 - assert any(isinstance(v, list) for k, v in md0.items() if k != "m") + assert any(isinstance(v, set) for k, v in md0.items() if k != "m") assert hits[1].id != "id0" assert hits[1].page_content == "[0, 1]" md1 = hits[1].metadata assert md1["m"] == 1 - assert any(isinstance(v, list) for k, v in md1.items() if k != "m") + assert any(isinstance(v, set) for k, v in md1.items() if k != "m")