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' )