def test_separate_comma_imports_none_to_separate(): input_partitions = [ CodePartition(CodeType.IMPORT, 'import os\n'), CodePartition(CodeType.NON_CODE, '\n'), CodePartition(CodeType.IMPORT, 'import six\n'), ] assert separate_comma_imports(input_partitions) == input_partitions
def test_separate_comma_imports_separates_some(): assert separate_comma_imports([ CodePartition(CodeType.IMPORT, 'import os, sys\n'), ]) == [ CodePartition(CodeType.IMPORT, 'import os\n'), CodePartition(CodeType.IMPORT, 'import sys\n'), ]
def test_separate_comma_imports_removes_comments(): # Since it's not really possible to know what the comma points to, we just # remove it assert separate_comma_imports([ CodePartition(CodeType.IMPORT, 'import os, sys # derp\n'), ]) == [ CodePartition(CodeType.IMPORT, 'import os\n'), CodePartition(CodeType.IMPORT, 'import sys\n'), ]
def test_separate_comma_imports_does_not_remove_comments_when_not_splitting(): input_partitions = [CodePartition(CodeType.IMPORT, 'import sys # noqa\n')] assert separate_comma_imports(input_partitions) == input_partitions
def test_separate_comma_imports_trivial(): assert separate_comma_imports([]) == []