fix flake8 errors

Browse Source
This commit is contained in:
Micha Gorelick 2020-03-13 15:05:07 +01:00
parent a9a15966e2
commit 03b7532997
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
test/test_dataset.py
View File

@ -396,7 +396,7 @@ class TableTestCase(unittest.TestCase):
def test_chunked_insert(self): def test_chunked_insert(self):
data = TEST_DATA * 100 data = TEST_DATA * 100
with chunked.ChunkedInsert(self.tbl) as chunk_tbl: with chunked.ChunkedInsert(self.tbl) as chunk_tbl:
for item in data: for item in data:
chunk_tbl.insert(item) chunk_tbl.insert(item)
assert len(self.tbl) == len(data) + 6 assert len(self.tbl) == len(data) + 6
@ -404,10 +404,11 @@ class TableTestCase(unittest.TestCase):
def test_chunked_insert_callback(self): def test_chunked_insert_callback(self):
data = TEST_DATA * 100 data = TEST_DATA * 100
N = 0 N = 0
def callback(queue): def callback(queue):
nonlocal N nonlocal N
N += len(queue) N += len(queue)
with chunked.ChunkedInsert(self.tbl, callback=callback) as chunk_tbl: with chunked.ChunkedInsert(self.tbl, callback=callback) as chunk_tbl:
for item in data: for item in data:
chunk_tbl.insert(item) chunk_tbl.insert(item)
assert len(data) == N assert len(data) == N