mirror of
https://github.com/simon987/hexlib.git
synced 2025-04-04 02:12:59 +00:00
37 lines
679 B
Python
37 lines
679 B
Python
from unittest import TestCase
|
|
|
|
from hexlib.misc import buffered
|
|
|
|
|
|
class TestBuffered(TestCase):
|
|
|
|
def test_simple(self):
|
|
my_list = []
|
|
|
|
@buffered(batch_size=2)
|
|
def put_item(items):
|
|
my_list.extend(items)
|
|
|
|
put_item([1, 2])
|
|
put_item([1])
|
|
put_item([1])
|
|
put_item([1])
|
|
|
|
self.assertEqual(len(my_list), 4)
|
|
|
|
def test_flush(self):
|
|
my_list = []
|
|
|
|
@buffered(batch_size=2)
|
|
def put_item(items):
|
|
my_list.extend(items)
|
|
|
|
put_item([1, 2])
|
|
put_item([1])
|
|
put_item([1])
|
|
put_item([1])
|
|
|
|
put_item(None)
|
|
|
|
self.assertEqual(len(my_list), 5)
|