flake8 fixes

This commit is contained in:
Simon Willison 2023-08-17 17:10:44 -07:00
commit 993029f466

View file

@ -93,12 +93,14 @@ def test_jsonsplit(fresh_db, delimiter):
],
pk="id",
)
fn = recipes.jsonsplit
if delimiter is not None:
def fn(value):
return recipes.jsonsplit(value, delimiter=delimiter)
else:
fn = recipes.jsonsplit
fresh_db["example"].convert("tags", fn)
assert list(fresh_db["example"].rows) == [
{"id": 1, "tags": '["foo", "bar"]'},
@ -121,11 +123,13 @@ def test_jsonsplit_type(fresh_db, type, expected):
],
pk="id",
)
fn = recipes.jsonsplit
if type is not None:
def fn(value):
return recipes.jsonsplit(value, type=type)
else:
fn = recipes.jsonsplit
fresh_db["example"].convert("records", fn)
assert json.loads(fresh_db["example"].get(1)["records"]) == expected