def test_add_statement_to_group_priority(self):
groups = ImportGroups()
groups.extend([RemainderGroup(), LocalGroup()])
groups.add_statement_to_group(ImportStatement([], ".a"))
self.assertListEqual(groups[0].statements, [])
self.assertListEqual(groups[1].statements, [ImportStatement([], ".a")])
def test_add_statement_to_group_priority(self):
groups = ImportGroups()
groups.extend([RemainderGroup(), LocalGroup()])
groups.add_statement_to_group(ImportStatement([], ".a"))
self.assertListEqual(groups[0].statements, [])
self.assertListEqual(groups[1].statements, [ImportStatement([], ".a")])
def test_add_statement_to_group_one(self):
groups = ImportGroups()
groups.extend([LocalGroup()])
with self.assertRaises(ValueError):
groups.add_statement_to_group(ImportStatement([], 'a'))
groups.add_statement_to_group(ImportStatement([], '.a'))
self.assertListEqual(groups[0].statements, [ImportStatement([], '.a')])
def test_add_statement_to_group_one(self):
groups = ImportGroups()
groups.extend([LocalGroup()])
with self.assertRaises(ValueError):
groups.add_statement_to_group(ImportStatement([], "a"))
groups.add_statement_to_group(ImportStatement([], ".a"))
self.assertListEqual(groups[0].statements, [ImportStatement([], ".a")])
def test_formatted_with_artifacts(self):
artifacts = {'sep': '\r\n'}
groups = ImportGroups(file_artifacts=artifacts)
groups.extend([
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')
def test_formatted_with_artifacts(self):
artifacts = {"sep": "\r\n"}
groups = ImportGroups(file_artifacts=artifacts)
groups.extend([
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")
def test_formatted_with_artifacts(self):
artifacts = {"sep": "\r\n"}
groups = ImportGroups(file_artifacts=artifacts)
groups.extend(
[
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"
)
