def METHOD_NAME(self, ds, documents):
    ds.write_documents(documents)
    bm25_representation = ds.bm25[ds.index]
    assert isinstance(bm25_representation, BM25)
    assert bm25_representation.corpus_size == ds.get_document_count()