mirror of
https://github.com/simon987/hexlib.git
synced 2025-12-13 22:59:04 +00:00
Add way to manually flush @buffered
This commit is contained in:
36
test/test_buffered.py
Normal file
36
test/test_buffered.py
Normal file
@@ -0,0 +1,36 @@
|
||||
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)
|
||||
Reference in New Issue
Block a user