From 993029f4669ae1e57862bd3576e1e3fd77477c5c Mon Sep 17 00:00:00 2001 From: Simon Willison Date: Thu, 17 Aug 2023 17:10:44 -0700 Subject: [PATCH] flake8 fixes --- tests/test_recipes.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/test_recipes.py b/tests/test_recipes.py index afb3e1a..eca3987 100644 --- a/tests/test_recipes.py +++ b/tests/test_recipes.py @@ -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