From 47531166cb6935993aa7916eb1ab2c8f27c9f211 Mon Sep 17 00:00:00 2001
From: Prodesire <prodesire@yeah.net>
Date: Sat, 3 Feb 2018 00:02:49 +0800
Subject: [PATCH] add test for iter.first and iter.last

---
 tests/test_iter.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)
 create mode 100644 tests/test_iter.py

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