def test_no_remainders(self):
     batched = tuple(textindex.batch([1, 2], batch_size=2))
     assert batched == ((1, 2),)
     batched = tuple(textindex.batch([1, 2, 3], batch_size=3))
     assert batched == ((1, 2, 3),)
     batched = tuple(textindex.batch([1, 2, 3, 4], batch_size=2))
     assert batched == ((1, 2), (3, 4))
     batched = tuple(textindex.batch([1, 2, 3, 4, 5, 6], batch_size=3))
     assert batched == ((1, 2, 3), (4, 5, 6))
Exemple #2
0
 def test_no_remainders(self):
     batched = tuple(textindex.batch([1, 2], batch_size=2))
     assert batched == ((1, 2), )
     batched = tuple(textindex.batch([1, 2, 3], batch_size=3))
     assert batched == ((1, 2, 3), )
     batched = tuple(textindex.batch([1, 2, 3, 4], batch_size=2))
     assert batched == ((1, 2), (3, 4))
     batched = tuple(textindex.batch([1, 2, 3, 4, 5, 6], batch_size=3))
     assert batched == ((1, 2, 3), (4, 5, 6))
 def test_single(self):
     for batch_size in range(1, 10):
         batched = tuple(textindex.batch([1], batch_size=batch_size))
         assert batched == ((1,),)
 def test_empty(self):
     for batch_size in range(1, 10):
         assert len(list(textindex.batch([], batch_size=batch_size))) == 0
Exemple #5
0
 def test_single(self):
     for batch_size in range(1, 10):
         batched = tuple(textindex.batch([1], batch_size=batch_size))
         assert batched == ((1, ), )
Exemple #6
0
 def test_empty(self):
     for batch_size in range(1, 10):
         assert len(list(textindex.batch([], batch_size=batch_size))) == 0