def test_add_statement_to_group_priority(self):
groups = ImportGroups()
groups.groups = [
RemainderGroup(),
LocalGroup(),
]
groups.add_statement_to_group(
ImportStatement([], '.a')
)
self.assertListEqual(
groups.groups[0].statements,
[]
)
self.assertListEqual(
groups.groups[1].statements,
[ImportStatement([], '.a')]
)
def test_add_statement_to_group_one(self):
groups = ImportGroups()
groups.groups = [
LocalGroup()
]
with self.assertRaises(ValueError):
groups.add_statement_to_group(
ImportStatement([], 'a')
)
groups.add_statement_to_group(
ImportStatement([], '.a')
)
self.assertListEqual(
groups.groups[0].statements,
[ImportStatement([], '.a')]
)
def test_formatted_with_artifacts(self):
artifacts = {'sep': '\r\n'}
groups = ImportGroups(file_artifacts=artifacts)
groups.groups = [
RemainderGroup(file_artifacts=artifacts),
LocalGroup(file_artifacts=artifacts),
]
groups.add_statement_to_group(
ImportStatement([], '.a', file_artifacts=artifacts)
)
groups.add_statement_to_group(
ImportStatement([], 'foo', file_artifacts=artifacts)
)
self.assertEqual(
groups.formatted(),
'import foo\r\n'
'\r\n'
'import .a'
)
