Improved tests for sum

diff --git a/tests/test_asyncfilters.py b/tests/test_asyncfilters.py
index 8732e78020ff27b332ab6fe860756691edb540b3..0fdef48114a233d23fdb10bf337bbe1b69b5d88f 100644 (file)
--- a/tests/test_asyncfilters.py
+++ b/tests/test_asyncfilters.py
@@ -184,18 +184,20 @@ def test_empty_map(env_async):
     assert tmpl.render() == '[]'
 
 
-def test_sum(env_async):
-    tmpl = env_async.from_string('''{{ [1, 2, 3, 4, 5, 6]|sum }}''')
-    assert tmpl.render() == '21'
+@mark_dualiter('items', lambda: [1, 2, 3, 4, 5, 6])
+def test_sum(env_async, items):
+    tmpl = env_async.from_string('''{{ items()|sum }}''')
+    assert tmpl.render(items=items) == '21'
 
 
-def test_sum_attributes(env_async):
-    tmpl = env_async.from_string('''{{ values|sum('value') }}''')
-    assert tmpl.render(values=[
-        {'value': 23},
-        {'value': 1},
-        {'value': 18},
-    ]) == '42'
+@mark_dualiter('items', lambda: [
+    {'value': 23},
+    {'value': 1},
+    {'value': 18},
+])
+def test_sum_attributes(env_async, items):
+    tmpl = env_async.from_string('''{{ items()|sum('value') }}''')
+    assert tmpl.render(items=items)
 
 
 def test_sum_attributes_nested(env_async):