add test for index in iteration context Refs amyreese#12

graingert · May 18, 2020 · c6494a4 · c6494a4
1 parent 4346ab2
commit c6494a4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/fissix/tests/test_fixers.py b/fissix/tests/test_fixers.py
@@ -3158,6 +3158,10 @@ def test_map_trailers(self):
         a = """x = list(map(f, 'abc'))[0][1]"""
         self.check(b, a)
 
+        b = """x = set(map(f, 'abc')[0])"""
+        b = """x = set(list(map(f, 'abc'))[0])"""
+        self.check(b, a)
+
     def test_trailing_comment(self):
         b = """x = map(f, 'abc')   #   foo"""
         a = """x = list(map(f, 'abc'))   #   foo"""