def METHOD_NAME(self, ds, documents):
    documents = [doc for doc in documents if doc.embedding is not None]
    ds.write_documents(documents)
    query_embs = [doc.embedding for doc in documents]
    docs_batch = ds.query_by_embedding_batch(query_embs=query_embs, top_k=5)
    assert len(docs_batch) == 6
    for docs, query_emb in zip(docs_batch, query_embs):
        assert len(docs) == 5
        assert (docs[0].embedding == query_emb).all()