Method test_with_sparse_vector_fields

python/tests/test_convert.py:184–218

(self)

            assert text_vector[i] == doc.vectors["text"][i]

    def test_with_sparse_vector_fields(self):
        schema = CollectionSchema(
            name="test_collection",
            vectors=[
                VectorSchema(
                    name="author",
                    data_type=DataType.SPARSE_VECTOR_FP32,
                ),
                VectorSchema(
                    name="content",
                    data_type=DataType.SPARSE_VECTOR_FP16,
                ),
            ],
        )
        doc = Doc(
            id="1",
            vectors={
                "author": {1: 1.1, 2: 2.2, 3: 3.3},
                "content": {4: 4.4, 5: 5.5, 6: 6.6},
            },
        )

        cpp_doc = convert_to_cpp_doc(doc, collection_schema=schema)
        assert cpp_doc is not None
        assert cpp_doc.pk() == doc.id

        author_vector = cpp_doc.get_any("author", DataType.SPARSE_VECTOR_FP32)
        assert isinstance(author_vector, dict)
        for key, value in doc.vector("author").items():
            assert math.isclose(author_vector[key], value, rel_tol=1e-1)

        content_vector = cpp_doc.get_any("content", DataType.SPARSE_VECTOR_FP16)
        assert isinstance(content_vector, dict)
        for key, value in doc.vector("content").items():
            assert math.isclose(content_vector[key], value, rel_tol=1e-1)

    def test_with_scalar_fields_error_datatype(self):
        schema = CollectionSchema(