def METHOD_NAME(self):
    for i in range(1, 11):  # try different batch sizes
        result = self.sbert.embed_batches(self.corpus.documents, i)
        self.assertEqual(result, IDEAL_RESPONSE[:-1] + [None])