o h @sdZddlmZddlmZddlmZddlmZdZdZ dZ dd ee d ed ee d ee fddZ d ed efddZ ddee dee dee d ee fddZdee de d ee fddZd S)zUtilities for truncating assertion output. Current default behaviour is to truncate assertion explanations at ~8 terminal lines, unless running in "-vv" mode or running on CI. )List)Optional)util)Itemizuse '-vv' to showN explanationitem max_lengthreturncCst|rt|S|S)zGTruncate this assertion explanation if the given test item is eligible.)_should_truncate_item_truncate_explanation)rrr r P/var/www/html/ai/venv/lib/python3.10/site-packages/_pytest/assertion/truncate.pytruncate_if_requiredsrcCs|jjj}|dko t S)z9Whether or not this test item is eligible for truncation.)configoptionverboser running_on_ci)rrr r rr s r input_lines max_lines max_charsc Cs|durt}|dur t}td|}|d}|d}t||kr'||kr'|S|d|}d}td||kr>t||}nd}t|t|}|dr[|dd|d<|rZ|d 7}nd|d<|dd |d |d krkdnd d tgS)aTruncate given list of strings that makes up the assertion explanation. Truncates to either 8 lines, or 640 characters - whichever the input reaches first, taking the truncation explanation into account. The remaining lines will be replaced by a usage message. NFrTFz...z...Full output truncated (z linesz hidden), )DEFAULT_MAX_LINESDEFAULT_MAX_CHARSlenjoin_truncate_by_char_count USAGE_MSG) rrrinput_char_counttolerable_max_charstolerable_max_linestruncated_explanationtruncated_chartruncated_line_countr r rr !sB    r cCspd}t|D]\}}|t||krn|t|7}q|d|}||}|r1||}|d|}|||S)Nr) enumeraterappend)rriterated_char_countiterated_index input_linetruncated_result final_linefinal_line_truncate_pointr r rr!ds   r!)N)NN)__doc__typingrr_pytest.assertionr _pytest.nodesrrrr"strintrboolr r r!r r r rs>       "C