From d4d236f14fe027c3f681db31a953de11ee916240 Mon Sep 17 00:00:00 2001 From: Andrew Kesterson Date: Tue, 31 Dec 2019 06:30:07 +0000 Subject: [PATCH] Fix failing filter test --- tests/test_broken_afilter.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/tests/test_broken_afilter.py b/tests/test_broken_afilter.py index 07db882..a59454e 100644 --- a/tests/test_broken_afilter.py +++ b/tests/test_broken_afilter.py @@ -36,8 +36,10 @@ def afilter(x): assert("d" not in dpath.util.search(dict, ['**'], afilter=afilter)['a']['b']['c']) def filter(x): - sys.stderr.write(x) - return x.get('type', None) == 'correct' + sys.stderr.write(str(x)) + if hasattr(x, 'get'): + return x.get('type', None) == 'correct' + return False a = { 'actions': [ @@ -51,9 +53,6 @@ def filter(x): } results = [[x[0], x[1]] for x in dpath.util.search(a, 'actions/*', yielded=True)] - print(results) results = [[x[0], x[1]] for x in dpath.util.search(a, 'actions/*', afilter=filter, yielded=True)] - print(filter) - print(results) assert(len(results) == 1) - assert(results[0]['type'] == 'correct') + assert(results[0][1]['type'] == 'correct')