tests/test_embedder.py

import numpy as np

from backend.app.core.embedder import _to_python_float_vectors


def test_embedding_vectors_are_plain_python_floats_for_chroma_validation():
    vectors = _to_python_float_vectors([[np.float32(0.25), np.float64(0.75)]])

    assert vectors == [[0.25, 0.75]]
    assert type(vectors[0][0]) is float
    assert type(vectors[0][1]) is float