gL i4\ddlZddlmZejdddlmZGddZy)N)Series matplotlibget_standard_colorsc eZdZejj ddgdfdgdfdgdfdd d gfd d gfgd Zejj dd d gfdgdfdgdfdgdfdgdfgdZejj dd dgfdgdfdgdfgdZejj dd gdfdgdfdgdfdgdfgdZ ejj dd gdfdgdfdgdfdgdfdgdfdgd fgd!Z ejj dd d"gfdd"d"gfdgd#fgd$Z ejj d%d&d d&gfd'dd'd'gfd(dgd)fgd*Z ejj d+d,gd-e gd./gd0Zejj d+gd1d2Zy3)4TestGetStandardColorsznum_colors, expectedredgreenblue)r r r r r )r r r r r r r r r cddl}ddlm}d|gdi}|j|5t |}||k(sJ dddy#1swYyxYw)Nrcycleraxes.prop_cycler colorrc num_colorsrmatplotlib.pyplotr rc_contextrselfrexpectedmplr mpl_paramsresults f/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/plotting/test_style.py)test_default_colors_named_from_prop_cyclez?TestGetStandardColors.test_default_colors_named_from_prop_cycle s[ !, v,DE ^^z^ * &(J?FX% %% & & &s AAb)r'gr)r'r(r)y)r'r(r)r+r')r'r(r)r+r'r(r)cddl}ddlm}d|di}|j|5t |}||k(sJ dddy#1swYyxYw)Nrrrbgryrrrrrs r%0test_default_colors_named_from_prop_cycle_stringzFTestGetStandardColors.test_default_colors_named_from_prop_cycle_stringsZ !, vF3 ^^z^ * &(J?FX% %% & & &s AAznum_colors, expected_nameC0)r/C1C2 ) r/r0r1C3C4C5C6C7C8C9r/r0cddl}ddlm}|ji5|Dcgc]}|j |}}t |}||k(sJ dddycc}w#1swYyxYw)Nrrr)rmatplotlib.colorscolorsrto_hexr)r r expected_namer"mcolorsxr!r$s r%.test_default_colors_named_undefined_prop_cyclezDTestGetStandardColors.test_default_colors_named_undefined_prop_cycle4si2 !+ ^^r^ " &3@Aaq)AHA(J?FX% %% & &A & &sAAAAA'r r 皙?皙?333333?)r r rCr c4gd}t||}||k(sJy)NrBrrrr rr!rr$s r%test_user_input_color_sequencez4TestGetStandardColors.test_user_input_color_sequenceUs#2$5ZH!!!)r)r(r'k)r)r(r'rLr))r)r(r'rLr)r(c0d}t||}||k(sJy)NrgbkrHrrIs r%test_user_input_color_stringz2TestGetStandardColors.test_user_input_color_stringcs#$5ZH!!!rKrC)rCrCrCc0d}t||}||k(sJy)NrCrHrrIs r%test_user_input_color_floatsz2TestGetStandardColors.test_user_input_color_floatsss# $5ZH!!!rKzcolor, num_colors, expectedCrimson DodgerBlue firebrick)rUrUrUc,t||}||k(sJy)NrHr)r rrr!r$s r%"test_user_input_named_color_stringz8TestGetStandardColors.test_user_input_named_color_strings%5ZH!!!rKrobject)dtypec~tjtd5t|ddddy#1swYyxYw)NzInvalid color argumentmatchrrHpytestraises ValueErrorrr rs r%test_empty_color_raisesz-TestGetStandardColors.test_empty_color_raisess0 ]]:-E F ; e : ; ; ;3<) bad_color)r r rf)rD)rDrE)rDrErFg?g?c~tjtd5t|ddddy#1swYyxYw)Nz Invalid colorr]rrHr_rcs r%test_bad_color_raisesz+TestGetStandardColors.test_bad_color_raisess1]]:_ = ; e : ; ; ;reN)__name__ __module__ __qualname__r`mark parametrizer&r.rArJrPrRrWrrdrhrYrKr%rr s [[ ( ) 8 9 G H  ! L    &  & [[ J  $ % ) * 3 4    &  & [[# K " #   0&10& [[ 1 2 1 2 1 2 8 9  ""  [[ $ % $ % $ % $ % ) * . /   " "  [[ ! " /2 3 C D ""  [[% I; ' 1|\: ; !D E "" [[Wr2r6"H3M&NO;P; [[  ; ;rKr)r`pandasr importorskip!pandas.plotting._matplotlib.stylerrrYrKr%rqs* L!AT;T;rK