Add Testcases

j30ng · j30ng · commit 048ec3e0ee97 · 2019-10-04T03:33:47.000+09:00
diff --git a/tests/snippets/stdlib_itertools.py b/tests/snippets/stdlib_itertools.py
@@ -243,3 +243,34 @@ def assert_matches_seq(it, seq):
 assert list(itertools.compress("ABCDEF", [1,0,1])) == list("AC")
 assert list(itertools.compress("ABC", [0,1,1,1,1,1])) == list("BC")
 assert list(itertools.compress("ABCDEF", [True,False,"t","",1,9])) == list("ACEF")
+
+
+# itertools.tee
+t = itertools.tee([])
+assert len(t) == 2
+assert t[0] is not t[1]
+assert list(t[0]) == list(t[1]) == []
+
+with assert_raises(TypeError):
+    itertools.tee()
+
+t = itertools.tee(range(1000))
+assert list(t[0]) == list(t[1]) == list(range(1000))
+
+t = itertools.tee([1,22,333], 3)
+assert len(t) == 3
+assert 1 == next(t[0])
+assert 1 == next(t[1])
+assert 22 == next(t[0])
+assert 333 == next(t[0])
+assert 1 == next(t[2])
+assert 22 == next(t[1])
+assert 333 == next(t[1])
+assert 22 == next(t[2])
+with assert_raises(StopIteration):
+    next(t[0])
+assert 333 == next(t[2])
+with assert_raises(StopIteration):
+    next(t[2])
+with assert_raises(StopIteration):
+    next(t[1])
