if member_name in (u'imap', u'izip', u'ifilter'):
child.value = None
child.remove()
- elif member_name == u'ifilterfalse':
+ elif member_name in (u'ifilterfalse', u'izip_longest'):
node.changed()
- name_node.value = u'filterfalse'
+ name_node.value = (u'filterfalse' if member_name[1] == u'f'
+ else u'zip_longest')
# Make sure the import statement is still sane
children = imports.children[:] or [imports]