P1iaKdZddlmZmZddlmZddlmZmZddl m Z m Z ddl m Z ddlmZddlmZdd lmZdd lmZGd d eZGd deZy)u Unit tests for tongshu views YBP (Yellow/Black Path) functionality. These tests verify that the tongshu calendar views correctly classify Yellow/Black Path gods as either good (黄道) or bad (黑道). Good gods (黄道): 青龙, 明堂, 金匮, 天德, 玉堂, 司命 Bad gods (黑道): 天刑, 朱雀, 白虎, 天牢, 玄武, 勾陈 Tests cover: - transform_day_data function classification - get_day_data function classification - month_view response data - specific known date classifications - consistency across different calculation methods - all 12 gods have correct classification Usage: python manage.py test tongshu.tests.TongshuYBPViewsTestCase python manage.py test_ybp_views # custom management command )TestCaseClient)reverse)datetime)transform_day_data get_day_data)TongshuCalendar)bz)test_safe_print)get_user_model)Personc:eZdZdZdZdZdZdZdZdZ dZ y ) TongshuYBPViewsTestCasezFTest cases for YBP (Yellow/Black Path) functionality in tongshu views.ct|_gd|_gd|_|j|jz|_gd|_y)zSet up test fixtures.)u青龙u明堂金匮u天德玉堂u司命)u天刑朱雀u白虎u天牢u玄武u勾陈))r)rr)rr)r )r)r)r N)rclient good_godsbad_godsall_gods test_dates)selfs //home/cursorai/projects/iching/tongshu/tests.pysetUpzTongshuYBPViewsTestCase.setUp%s9h VT 6  c tjdd}|ddd}d}d}|D]%}t|dd}|jd||d}|jd||jd ||jd ||d}|j||jd ||d } |j| d d gd| ||j vr |j | d d|d| d|dz }n-||jvr|j | d d|d| d|dz }|d } |j| dd|j| dd(|j|dd|j|ddtd|d|dy)zJTest that transform_day_data correctly classifies YBP gods as good or bad.rrNrryellow_black_pathgodtypepositionUnknown YBP god: goodbadInvalid YBP type: Good god & should be classified as 'good', got ''Bad god % should be classified as 'bad', got 'zYBP position should be >= 1rzYBP position should be <= 12(Should have tested at least one good god'Should have tested at least one bad god ✅ Tested  good gods and z bad gods in transform_day_data) r get_month_datarassertInr$r" assertEqualr#assertGreaterEqualassertLessEqual assertGreaterr ) r& calendar_data sample_days good_count bad_count day_array transformedybp_datagod_nameybp_typer/s r'*test_transform_day_data_ybp_classificationzBTongshuYBPViewsTestCase.test_transform_day_data_ybp_classification9s(66tQ? #A&s+   $I,Ya@K MM-{ ;"#67H MM% * MM&( + MM*h / H MM(DMM5Fxj3Q R 'H MM(VUO7I(5T U4>>)  6z)OPXzYZ[]a T]]*  5xj(MhZWXY[Q  +H  # #Ha1N O  2/M NC%H :q*TU 9a)RS+j\ Kjklr)c d}d}|jD]\}}}t|||}t|}|jd||d}|jd||jd||jd||d} |j| |jd| |d} |j| ddgd | | |j vr"|j | dd | d | d ||d z }| |jvs|j | dd| d| d ||d z }|j|dd|j|ddtd|d|dy)zDTest that get_day_data correctly classifies YBP gods as good or bad.rr,r-r.r/r0r1r2r3r4r5z' for rr7r8r9r:r;r<z bad gods in get_day_dataN) r%rr r>r$r"r?r#rBr ) r&rErFyearmonthdaydate_objday_datarIrJrKs r'$test_get_day_data_ybp_classificationz>)  6z)OPXzY_`h_ijla T]]*  5xj(MhZW]^f]ghjQ 9!0> :q*TU 9a)RS+j\ Kdefr)c tdddd}|jj|}|j|jd|j d|j |j d}|jt|dd d}d}d}|D]}|D]}|dz }|j d ||d } |j d | |j d | |j d | | d } | d } |j | |jd| |j | ddgd| | dk(r'|dz }|j | |jd| d| dk(s|dz }|j | |jd| d|j|dd|j|dd|j|ddtd|d|d|dy)zHTest that the month view response contains properly classified YBP data.tongshu:month_viewrrrNrOkwargsweeksrzShould have at least one weekr,r-r.r/r0r1r2r3zGod z/ classified as 'good' but not in good_gods listz- classified as 'bad' but not in bad_gods listz#Should have tested at least 30 daysz!Should have at least one good dayz Should have at least one bad dayu✅ Month view: Tested z days with z good and z bad classificationsN) rr!getr? status_coder>contextrBlenr$r"r#r ) r&urlresponserZrErF total_daysweekrPrIrJrKs r'*test_month_view_response_contains_ybp_datazBTongshuYBPViewsTestCase.test_month_view_response_contains_ybp_datas*D13MN;;??3' --s3 gx//0  ) 3u:q*IJ   Da  13723 eX. fh/ j(3$E?#F+ h 9J8*7UV h;MhZ9XYv%!OJMM(DNNxj(WXZ&NIMM(DMMxj(UVX3< :r+PQ :q*MN 9a)KL1*[ T^_h^ii}~r)cvtdddddftdddddftdddddftddd d d fg}|D]\}}}|j||| 5t|}|d }|j|d|d|d|d|d|j|d|d|d|d|d|dt j |j |j|j}|j||j|d||j|d|dddy#1swYxYw)z3Test specific dates with known YBP classifications.rrrr1rrrrrrr2)r expected_god expected_typer,r-z Expected z for z, got r.z on N) rsubTestr r?r calcYellowBlackPathrNrOrPassertIsNotNone)r& known_cases test_daterfrgrRrI calc_results r''test_specific_known_ybp_classificationsz?TongshuYBPViewsTestCase.test_specific_known_ybp_classificationssp $1 x 0 $1 x 0 $1 x 0 $B 5 1  7B 2I|]9i[PVW_`fWgVhik!44Y^^Y__V_VcVcd $$[1  U!3\B  V!4mDfe7Bees CD//D8 c tddd}tj|j|j|j }t |}|d}tj|j|j}d}|jd}|dD]}|d|k(s |}n|j|d |d t||j|j} | d} |j|d |d d |j|d |d d|j|d | d d|j|d | d dtd|d|d d|d dy)zETest that YBP calculations are consistent across different functions.rrrr,Nz%Y-%m-%dr+rzCould not find z in calendar datar-z;Direct calculation and view function should return same godr.z J L %-*> I K &)=+@ J L :9+R TYHZG[[]^ijp^q]rrstur)c t}i}tddDcgc]}d|f}}|D]~\}}tj||}|dD][}t |||}|d} | d} | d} |j | | |vr| || <:|j || | d| d || d | ]|jt|d d t||jD]l\} } | |jvr|j | d d| d| d/| |jvr|j | dd| d| dX|jd| dnt|D cgc]} | |jvs| c} }t|D cgc]} | |jvs| c} }tdt|d|d|dtdt|D cgc]} | |jvs| c} tdt|D cgc]} | |jvs| c} ycc}wcc} wcc} wcc} wcc} w)zKTest that all 12 YBP gods have the correct classification when they appear.r rr+r,r-r.z Inconsistent classification for z : saw both z and z6Should have seen at least 8 different gods, only saw: r1r4z was classified as 'r6r2r7z Unknown god z encounteredr;z different gods: z good, z badz Good gods seen: z Bad gods seen: N)setranger r=raddr?r@r_sorteditemsr"r#failr )r& gods_seenclassifications_testedrO test_monthsrNrCrGrHrIrJrK observed_typeg good_seenbad_seens r'0test_all_twelve_gods_have_correct_classificationzHTongshuYBPViewsTestCase.test_all_twelve_gods_have_correct_classificationsE !#382,?,e}, ?&KD%+::4GM*1- 0D%H &':;#E?#F+ h'#997?*84$$%;H%Ex:8*KPfgoPpOqqvwwABC.'* IDVIEVDW X Z(>'C'C'E #Hm4>>)  z)=m_ANPT]]*  xj(<]O1MO L ,?@(FIEIqdnn1DIEF 9C9aT]]0B9CD+c)n%55FykQXYaXbbfgh-f5bAaSWSaSaNaa5b.c-def,V 4` 1QRVR_R_M_Q 4`-a,bcdU@JFC6c4`s5 H)H./H.H3H3H8#H8H=H=N) __name__ __module__ __qualname____doc__r(rLrSrdrnrvrr)r'rr"s2P (1mf(gT3@jE8'vR2er)rc6eZdZdZdZd dZdZdZdZdZ y) TongshuUserBaZiHeaderTestszSTests for the UI header that shows the logged-in user's BaZi in tongshu month view.c~t|_t}|jj dddd|_y)N 10000000001p1zu1@example.comU1)phonepasswordemail first_name)rr!r objects create_useruser)r&Users r'r(z TongshuUserBaZiHeaderTests.setUpHs4h LL,,=4Wgtx,y r)cdddddddddd}|rd d d|d <td td d dtdd|jd|}d|_|j |S)Nr)r-earthr+rrry)rPrOrNrrhourTesteri r[T)name birth_date birth_time created_byowner bazi_result)rrrr_skip_bazi_calculationsave)r&has_hourrps r'_create_owner_personz/TongshuUserBaZiHeaderTests._create_owner_personMs~q)+* *+a"8K  D!R(B|yy#  $(  r)c|jd}|j|ddd}|j||}|dk(r|dz}|||S)N 我的八字u&Header label 我的八字 should existz