def test_add_statement(self) -> None:
groups = ImportGroups([LocalGroup()])
with pytest.raises(ValueError):
groups.add_statement(ImportStatement("a"))
groups.add_statement(ImportStatement(".a"))
assert groups.groups[0].statements == [ImportStatement(".a")]
def test_formatted_with_artifacts(self) -> None:
artifacts = Artifacts(sep="\r\n")
groups = ImportGroups(
[
RemainderGroup(artifacts=artifacts),
LocalGroup(artifacts=artifacts)
],
artifacts=artifacts,
)
groups.add_statement(ImportStatement(".a"))
groups.add_statement(ImportStatement("foo"))
assert groups.formatted() == "import foo\r\n\r\nimport .a"
def test_add_statement_priority(self) -> None:
groups = ImportGroups([RemainderGroup(), LocalGroup()])
groups.add_statement(ImportStatement(".a"))
assert groups.groups[0].statements == []
assert groups.groups[1].statements == [ImportStatement(".a")]
