Class MockCollectionSchema

python/tests/test_query_executor.py:46–54

# Mock Collection Schema
# ----------------------------
class MockCollectionSchema(CollectionSchema):
    def __init__(self, vectors=Union[VectorSchema, Dict[str, VectorSchema]]):
        self._vectors = (
            [vectors] if not isinstance(vectors, Dict) else list(vectors.values())
        )

    @property
    def vectors(self):
        return self._vectors


# ----------------------------