diff options
Diffstat (limited to 'tests/test_util.py')
-rw-r--r-- | tests/test_util.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tests/test_util.py b/tests/test_util.py index e69de29..c2428e8 100644 --- a/tests/test_util.py +++ b/tests/test_util.py @@ -0,0 +1,9 @@ +from mlscraper.util import no_duplicates_generator_decorator + + +def test_no_duplicates_generator_decorator(): + @no_duplicates_generator_decorator + def decorated_generator(): + yield from [1, 1, 2, 3, 3, 3] + + assert list(decorated_generator()) == [1, 2, 3] |