def test_all_line_numbers(self): groups = ImportGroups() self.assertListEqual(groups.all_line_numbers(), []) g = BaseImportGroup() g.statements = [mock.MagicMock(line_numbers=[2, 7], spec=ImportStatement)] groups.groups.append(g) g = BaseImportGroup() g.statements = [mock.MagicMock(line_numbers=[1, 2], spec=ImportStatement)] groups.groups.append(g) self.assertListEqual(groups.all_line_numbers(), [1, 2, 7])
def test_all_line_numbers(self): s2 = ImportStatement([2, 7], 'b') s1 = ImportStatement([1, 2], 'a') group = BaseImportGroup() group.statements = [s1, s2] self.assertListEqual(group.all_line_numbers(), [1, 2, 7])
def test_all_line_numbers(self): s2 = ImportStatement([2, 7], "b") s1 = ImportStatement([1, 2], "a") group = BaseImportGroup() group.statements = [s1, s2] self.assertListEqual(group.all_line_numbers(), [1, 2, 7])
def test_as_string(self): group = BaseImportGroup() group.statements = [ImportStatement([], 'b'), ImportStatement([], 'a')] self.assertEqual( group.as_string(), 'import a\n' 'import b' )
def test_as_string_with_artifacts(self): group = BaseImportGroup(file_artifacts={'sep': '\r\n'}) group.statements = [ImportStatement([], 'b'), ImportStatement([], 'a')] self.assertEqual( group.as_string(), 'import a\r\n' 'import b' )
def test_merged_statements_special(self): group = BaseImportGroup() group.statements = [ImportStatement([], 'a', [ImportLeaf('*')]), ImportStatement([], 'b', [ImportLeaf('c')])] actual = group.merged_statements self.assertListEqual(sorted(actual), [ ImportStatement([], 'a', [ImportLeaf('*')]), ImportStatement([], 'b', [ImportLeaf('c')]), ])
def test_formatted(self): group = BaseImportGroup() group.statements = [ ImportStatement([], 'b' * 80, [ImportLeaf('c'), ImportLeaf('d')]), ImportStatement([], 'a') ] self.assertEqual( group.formatted(), 'import a\n' + 'from {} import (\n'.format('b' * 80) + ' c,\n' + ' d,\n' + ')')
def test_formatted(self): group = BaseImportGroup() group.statements = [ ImportStatement([], "b" * 80, [ImportLeaf("c"), ImportLeaf("d")]), ImportStatement([], "a"), ] self.assertEqual( group.formatted(), "import a\n" + "from {} import (\n".format("b" * 80) + " c,\n" + " d,\n" + ")", )
def test_unique_statements(self, mock_set, mock_list, mock_sorted): group = BaseImportGroup() group.statements = mock.sentinel.statements actual = group.unique_statements self.assertEqual(actual, mock_sorted.return_value) mock_set.assert_called_once_with(mock.sentinel.statements) mock_list.assert_called_once_with(mock_set.return_value) mock_sorted.assert_called_once_with(mock_list.return_value)
def test_formatted_with_artifacts(self): artifacts = {'sep': '\r\n'} group = BaseImportGroup(file_artifacts=artifacts) group.statements = [ ImportStatement(list(), 'b' * 80, [ImportLeaf('c'), ImportLeaf('d')], file_artifacts=artifacts), ImportStatement([], 'a', file_artifacts=artifacts) ] self.assertEqual( group.formatted(), 'import a\r\n' + 'from {} import (\r\n'.format('b' * 80) + ' c,\r\n' + ' d,\r\n' + ')')
def test_merged_statements_special(self): group = BaseImportGroup() group.statements = [ ImportStatement([], "a", [ImportLeaf("*")]), ImportStatement([], "b", [ImportLeaf("c")]), ] actual = group.merged_statements self.assertListEqual( sorted(actual), [ ImportStatement([], "a", [ImportLeaf("*")]), ImportStatement([], "b", [ImportLeaf("c")]), ], )
def test_formatted(self): group = BaseImportGroup() group.statements = [ ImportStatement([], 'b' * 80, [ImportLeaf('c'), ImportLeaf('d')]), ImportStatement([], 'a') ] self.assertEqual( group.formatted(), 'import a\n' + 'from {} import (\n'.format('b' * 80) + ' c,\n' + ' d,\n' + ')' )
def test_formatted_with_artifacts(self): artifacts = {'sep': '\r\n'} group = BaseImportGroup(file_artifacts=artifacts) group.statements = [ ImportStatement(list(), 'b' * 80, [ImportLeaf('c'), ImportLeaf('d')], file_artifacts=artifacts), ImportStatement([], 'a', file_artifacts=artifacts) ] self.assertEqual( group.formatted(), 'import a\r\n' + 'from {} import (\r\n'.format('b' * 80) + ' c,\r\n' + ' d,\r\n' + ')' )
def test_formatted_with_artifacts(self): artifacts = {"sep": "\r\n"} group = BaseImportGroup(file_artifacts=artifacts) group.statements = [ ImportStatement( list(), "b" * 80, [ImportLeaf("c"), ImportLeaf("d")], file_artifacts=artifacts, ), ImportStatement([], "a", file_artifacts=artifacts), ] self.assertEqual( group.formatted(), "import a\r\n" + "from {} import (\r\n".format("b" * 80) + " c,\r\n" + " d,\r\n" + ")", )
def test_as_string(self): group = BaseImportGroup() group.statements = [ImportStatement([], "b"), ImportStatement([], "a")] self.assertEqual(group.as_string(), "import a\n" "import b")
def test_as_string_with_artifacts(self): group = BaseImportGroup(file_artifacts={"sep": "\r\n"}) group.statements = [ImportStatement([], "b"), ImportStatement([], "a")] self.assertEqual(group.as_string(), "import a\r\n" "import b")