Test sync and async generator

1 files changed, 8 insertions, 6 deletions

diff --git a/tests/test_asyncfilters.py b/tests/test_asyncfilters.py
index 69f4e92..c5c55b7 100644
--- a/tests/test_asyncfilters.py
+++ b/tests/test_asyncfilters.py
@@ -118,13 +118,15 @@ def test_join_attribute(env_async, users):
     assert tmpl.render(users=users) == 'foo, bar'
 
 
-def test_simple_reject(env_async):
-    tmpl = env_async.from_string('{{ [1, 2, 3, 4, 5]|reject("odd")|join("|") }}')
-    assert tmpl.render() == '2|4'
+@mark_dualiter('items', lambda: [1, 2, 3, 4, 5])
+def test_simple_reject(env_async, items):
+    tmpl = env_async.from_string('{{ items()|reject("odd")|join("|") }}')
+    assert tmpl.render(items=items) == '2|4'
 
 
-def test_bool_reject(env_async):
+@mark_dualiter('items', lambda: [None, False, 0, 1, 2, 3, 4, 5])
+def test_bool_reject(env_async, items):
     tmpl = env_async.from_string(
-        '{{ [none, false, 0, 1, 2, 3, 4, 5]|reject|join("|") }}'
+        '{{ items()|reject|join("|") }}'
     )
-    assert tmpl.render() == 'None|False|0'
+    assert tmpl.render(items=items) == 'None|False|0'