o h=@spdZddlmZddlmZddlmZddlmZdZ e dZ dZ dddZ dddZ dddZdddZdS) zUtilities for truncating assertion output. Current default behaviour is to truncate assertion explanations at terminal lines, unless running with an assertions verbosity level of at least 2 or running on CI. ) annotations)util)Config)ItemPzuse '-vv' to show explanation list[str]itemrreturncCs$t|\}}}|rt|||dS|S)zGTruncate this assertion explanation if the given test item is eligible.) max_lines max_chars)_get_truncation_parameters_truncate_explanation)rr should_truncater r r[/var/www/html/openai_agents/venv/lib/python3.10/site-packages/_pytest/assertion/truncate.pytruncate_if_requiredsrtuple[bool, int, int]cCs~|jd}t|dur |nt}|jd}t|dur|nt}|jtj}|dko/t }|o9|dkp9|dk}|||fS)zgReturn the truncation parameters related to the given item, as (should truncate, max lines, max chars).truncation_limit_linesNtruncation_limit_charsr) configgetiniintDEFAULT_MAX_LINESDEFAULT_MAX_CHARS get_verbosityrVERBOSITY_ASSERTIONSr running_on_ci)r r r verboserrrrrs   r input_linesr rr c Cstd|}|d}|d}t||kr||kr|S|dkr&|d|}n|}d}td||kr=|dkr=t||}nd}||krE|St|t|}|dr`|dd |d<|r_|d 7}nd |d<g|dd |d |d krsdnd dtS)aTruncate given list of strings that makes up the assertion explanation. Truncates to either max_lines, or max_chars - whichever the input reaches first, taking the truncation explanation into account. The remaining lines will be replaced by a usage message. FrrNTFz...z...Full output truncated (z linesz hidden), )lenjoin_truncate_by_char_count USAGE_MSG) r!r r input_char_counttolerable_max_charstolerable_max_linestruncated_explanationtruncated_chartruncated_line_countrrrr4sH   rcCspd}t|D]\}}|t||krn|t|7}q|d|}||}|r1||}|d|}|||S)Nr) enumerater'append)r!r iterated_char_countiterated_index input_linetruncated_result final_linefinal_line_truncate_pointrrrr)zs   r)N)rr r rr r )r rr r)r!r r rr rr r )r!r r rr r )__doc__ __future__r_pytest.assertionr_pytest.configr _pytest.nodesrrrr*rrrr)rrrrs      F