def equivalent_source_variants__for_expression_parser_2( original_arguments: Arguments, ) -> List[NEA[Assertion[ParseSource], ParseSource]]: num_lines = original_arguments.num_lines return [ NEA(name='additional_arguments_on_same_line={}, additional_lines={}'. format( repr(following_arguments.first_line), repr(following_arguments.following_lines), ), expected=source_assertion, actual=original_arguments.last_line_followed_by( following_arguments, first_line_separator='').as_remaining_source) for following_arguments, source_assertion in _source_variants_with__for_expression_parser(num_lines) ]
def equivalent_source_variants__for_expression_parser( put: unittest.TestCase, original_arguments: Arguments, ) -> Iterator[ParseSource]: """ Assumes that the body of the loop parses using the given source. """ for following_arguments, source_assertion in _source_variants_with__for_expression_parser( original_arguments.num_lines): with put.subTest(additional_arguments_on_same_line=repr( following_arguments.first_line), additional_lines=repr( following_arguments.following_lines)): source = original_arguments.last_line_followed_by( following_arguments, first_line_separator='').as_remaining_source yield source source_assertion.apply_with_message(put, source, 'source after parse')