From 5ca400b689e1f71d518b6d87666ccc3930386652 Mon Sep 17 00:00:00 2001 From: Oliver Rice Date: Wed, 26 Mar 2025 14:35:49 -0500 Subject: [PATCH] add test cases for #105 --- src/tests/test_client.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/src/tests/test_client.py b/src/tests/test_client.py index 6e8694f..a4c8ec2 100644 --- a/src/tests/test_client.py +++ b/src/tests/test_client.py @@ -17,6 +17,18 @@ def test_create_collection(client: vecs.Client) -> None: client.create_collection(name="docs", dimension=384) +def test_get_or_create_collection(client: vecs.Client) -> None: + client.get_or_create_collection(name="resumes", dimension=1536) + # no error is raised + client.get_or_create_collection(name="resumes", dimension=1536) + + +def test_get_or_create_collection_dim_change(client: vecs.Client) -> None: + client.get_or_create_collection(name="resumes", dimension=1536) + with pytest.raises(vecs.exc.MismatchedDimension): + client.get_or_create_collection(name="resumes", dimension=1) + + def test_get_collection(client: vecs.Client) -> None: with pytest.warns(DeprecationWarning): with pytest.raises(vecs.exc.CollectionNotFound):