def test_removed(self) -> None:
import_statement = ImportStatement(
'hoge',
[
ImportAsPart('AAA', 'a'),
ImportAsPart('BBB', 'b'),
ImportAsPart('c', ''),
],
)
# remove as-imported name
removed = import_statement.removed('hoge.AAA as a')
assert removed is not None
# remove no-as-import name
removed = removed.removed('hoge.c')
assert removed is not None
# assert that one import is remained
assert len(removed.import_as_parts) == 1
assert removed.import_as_parts[0].name == 'b'
# remove the last one
removed = removed.removed('hoge.BBB as b')
assert removed is None
return
def test_repr(self) -> None:
import_statement = ImportStatement(
'hoge',
[
ImportAsPart('Hoge', 'hoge'),
ImportAsPart('Fuga', 'fuga'),
],
)
assert (str(import_statement) ==
'from hoge import Fuga as fuga, Hoge as hoge')
def test_of(self) -> None:
# named import
res = ImportAsPart.of('tensorflow as tf')
assert res is not None
assert res.import_what == 'tensorflow'
assert res.as_what == 'tf'
# non-named import
res = ImportAsPart.of('sys')
assert res is not None
assert res.import_what == 'sys'
assert res.as_what == ''
return
def test_merge(self) -> None:
# merge if from_what is the same
import_statement = ImportStatement(
'hoge',
[
ImportAsPart('AAA', 'a'),
ImportAsPart('BBB', 'b'),
],
)
target = ImportStatement('hoge', [
ImportAsPart('BBB', 'b'),
ImportAsPart('CCC', 'c'),
])
import_statement.merge(target)
assert len(import_statement.import_as_parts) == 3
# not merge if from_what is not the same
import_statement = ImportStatement(
'hoge',
[
ImportAsPart('AAA', 'a'),
ImportAsPart('BBB', 'b'),
],
)
target = ImportStatement('fuga', [
ImportAsPart('BBB', 'b'),
ImportAsPart('CCC', 'c'),
])
import_statement.merge(target)
assert len(import_statement.import_as_parts) == 2
def test_of_lines(self) -> None:
lines = [
'import os # tailing comment',
'# comment above',
'import sys',
'from pprint import ('
' pprint as pp,',
' pformat,',
')',
'from typing import (',
' List as L, # tailing comment 1',
' Dict as D,# tailing comment 2',
')',
'',
'# multi-line comment 1',
'# multi-line comment 2',
'import tensorflow as tf',
]
actuals = ImportStatement.of_lines(lines)
expecteds: List[ImportStatement] = [
ImportStatement(
'',
[
ImportAsPart('os', '', 'tailing comment'),
],
),
ImportStatement(
'',
[
ImportAsPart('sys', '', 'comment above'),
],
),
ImportStatement(
'pprint',
[
ImportAsPart('pprint', 'pp'),
ImportAsPart('pformat', ''),
],
),
ImportStatement(
'typing',
[
ImportAsPart('List', 'L', 'tailing comment 1'),
ImportAsPart('Dict', 'D', 'tailing comment 2'),
],
),
ImportStatement(
'',
[
ImportAsPart('tensorflow', 'tf', ('multi-line comment 1 '
'multi-line comment 2')),
],
),
]
assert actuals is not None
assert len(actuals) == len(expecteds)
assert all([str(a) == str(e)
for a, e in zip(actuals, expecteds)]) is True
return
def test_merge_list(self) -> None:
import_statements = [
ImportStatement('hoge', [
ImportAsPart('AAA', 'a'),
ImportAsPart('BBB', 'b'),
]),
ImportStatement('fuga', [
ImportAsPart('AAA', 'a'),
ImportAsPart('BBB', 'b'),
]),
ImportStatement('hoge', [
ImportAsPart('CCC', 'c'),
ImportAsPart('DDD', 'd'),
]),
]
actuals = ImportStatement.merge_list(import_statements)
expecteds = [
ImportStatement('hoge', [
ImportAsPart('AAA', 'a'),
ImportAsPart('BBB', 'b'),
ImportAsPart('CCC', 'c'),
ImportAsPart('DDD', 'd'),
]),
ImportStatement('fuga', [
ImportAsPart('AAA', 'a'),
ImportAsPart('BBB', 'b'),
]),
]
assert len(actuals) == len(expecteds)
assert all([str(a) == str(e)
for a, e in zip(actuals, expecteds)]) is True