def test_calculate_batchsize_maxlen():
""" Batch size and max length are calculated properly.
"""
texts = ['a b c d', 'e f g h i']
batch_size, maxlen = calculate_batchsize_maxlen(texts)
assert batch_size == 250
assert maxlen == 10, maxlen
def test_calculate_batchsize_maxlen():
""" Batch size and max length are calculated properly.
"""
texts = ['a b c d',
'e f g h i']
batch_size, maxlen = calculate_batchsize_maxlen(texts)
assert batch_size == 250
assert maxlen == 10, maxlen