Merge 3b97fa7 into 8d860c3

more-itertools · Oct 20, 2019 · 0c667f6 · 0c667f6
2 parents 8d860c3 + 3b97fa7
commit 0c667f6
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 #
 # more-itertools documentation build configuration file, created by
 # sphinx-quickstart on Mon Jun 25 20:42:39 2012.

diff --git a/more_itertools/more.py b/more_itertools/more.py
@@ -949,8 +949,7 @@ def walk(node, level):
             return
         else:
             for child in tree:
-                for x in walk(child, level + 1):
-                    yield x
+                yield from walk(child, level + 1)
 
     yield from walk(iterable, 0)
 

diff --git a/tests/test_more.py b/tests/test_more.py
@@ -518,7 +518,7 @@ def test_ilen(self):
         )
 
         # Empty
-        self.assertEqual(mi.ilen((x for x in range(0))), 0)
+        self.assertEqual(mi.ilen(x for x in range(0)), 0)
 
         # Iterable with __len__
         self.assertEqual(mi.ilen(list(range(6))), 6)
@@ -2692,7 +2692,7 @@ def test_repeated(self):
     def test_each_correct(self):
         a = set(range(6))
         for p in mi.set_partitions(a):
-            total = set(e for g in p for e in g)
+            total = {e for g in p for e in g}
             self.assertEqual(a, total)
 
     def test_duplicates(self):