def test_for_statement(self): content = ''' def for_s(word): for l in word: pass for i, letter in enumerate(word): yield i, letter ''' ret = extract_assignment(content) self.assertListEqual(ret, ['word', 'l', 'i', 'letter', ])
def test_with_statement(self): content = ''' def for_s(): with none(): pass with open('a') as f: f.read() with A() as e, B() as g: return e.read() ''' ret = extract_assignment(content) self.assertListEqual(ret, ['f', 'e', 'g', ])
def test_extruct_arguments(self): content = ''' def arg(a, b, c): return a, b, c''' ret = extract_assignment(content) self.assertEqual(ret, ['a', 'b', 'c', ])