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")]