diff --git a/tests/test_iter.py b/tests/test_iter.py new file mode 100644 index 0000000..de11ed9 --- /dev/null +++ b/tests/test_iter.py @@ -0,0 +1,15 @@ +import pytest +from pydu.iter import first, last + + +@pytest.mark.parametrize( + 'iterable', ( + [1, 2], + (1, 2), + {1, 2}, + {1: 1, 2: 2}, + iter([1, 2]) + )) +def test_first_last(iterable): + assert first(iterable) == 1 + assert last(iterable) == 2