P1ifLJdZddlZddlmZmZmZddlmZddlm Z ddl m Z ddl m Z m Z mZmZmZmZmZmZGdd ZGd d eZGd d eZGddeZGddeZGddeZGddeZGddeZGddeZGddeZy)z Test cases for LiuYao AI analysis functionality. This module tests various scenarios including normal cases, edge cases, and outliers. N)Mockpatch MagicMock)TestCase)timezone)datetime)LiuYaoConstants DataFormatterResponseValidatorget_hexagram_nameget_trigram_nameidentify_changing_linesanalyze_liuyaoprepare_liuyao_promptceZdZdZdZdZy) MockLiuYaozMock LiuYao object for testing.c |jdd|_|jdd|_|jdtj|_|jdd|_|jdd |_|jd d|_|jd d |_ |jd d|_ |jdd|_ |jdd|_ |jdi|_ |jdd|_|jdd|_|jdd|_y)Nidquestionu 测试问题qdatey1111y20y3y41y5y6readingdata ai_analysisanalysis_statuspendinganalysis_timestamp)getrrrnowrrrrrrr r!r#r$r%r'selfkwargss ?/home/cursorai/projects/iching/ai/tests/test_liuyao_analysis.py__init__zMockLiuYao.__init__s **T1% :~> ZZ8 **T5)**T3'**T5)**T3'**T5)**T5)zz)R0 JJvr* !::mT:%zz*;YG"(**-A4"Hc y)zMock save method.Nr*s r-savezMockLiuYao.save+s r/N)__name__ __module__ __qualname____doc__r.r2r1r/r-rrs)I  r/rc(eZdZdZdZdZdZdZy)TestLiuYaoConstantsz,Test LiuYao constants and utility functions.c|jttjdt ddD]"}|j |tj$y)z'Test that all 64 hexagrams are defined.@rAN) assertEquallenr HEXAGRAM_NAMESrangeassertInr+is r-test_hexagram_names_coveragez0TestLiuYaoConstants.test_hexagram_names_coverage3sA _;;rrR>rrdrerfrRN)rrr<set)r+liuyaochanging_liness r-test_identify_changing_linesz1TestUtilityFunctions.test_identify_changing_linesesu35UuU08 ^,o>ssssssK08 ,u5UuQVW08 ^,.@Ar/N)r3r4r5r6r^rarkr1r/r-rZrZMs! GCBr/rZc4eZdZdZdZdZdZdZdZdZ y) TestDataFormatterzTest DataFormatter class.cTtjdd}|j|dy)zTest formatting string data.z test stringtestNr format_datar<r+results r-test_format_data_stringz)TestDataFormatter.test_format_data_stringzs$**=&A /r/ctddd}tj|d}ddg}|D]}|j||y)z Test formatting dictionary data.value1value2)key1key2rou key1：value1u key2：value2Nr rqr@r+r#rsexpected_lineslines r-test_format_data_dictz'TestDataFormatter.test_format_data_dicts@ (3**48)?;"D MM$ '#r/crgd}tj|d}gd}|D]}|j||y)zTest formatting list data.)item1item2item3roNrzr{s r-test_format_data_listz'TestDataFormatter.test_format_data_lists6***484"D MM$ '#r/ctjdd}|j|dtjdd}|j|dy)zTest formatting empty data.Nrou无test数据 (No test data)r"rprrs r-test_format_data_emptyz(TestDataFormatter.test_format_data_emptysH**48 !?@**2v6 !?@r/cTtdD]L}tj|}|jd||jtj ||Ntjd}|j |dtjd}|jd|y)zTest Six Gods formatting.rRu六神:Nu"无六神数据 (No Six Gods data) u未知 (Unknown): 10)r?r format_six_godsr@r rWr<)r+rBrss r-test_format_six_godsz&TestDataFormatter.test_format_six_godssqA"2215F MM)V , MM/77:F C ..t4 !EF..r2 ,f5r/clddgddgd}tj|}|jd||jd||jd||jd |tji}|j|d tjd }|j|d y ) zTest wangshui line formatting.emptyzcs12-3producezcs12-7)daymonthu日:u月:u空u生r"N)r format_wangshui_liner@r<)r+rel_datarss r-test_format_wangshui_linez+TestDataFormatter.test_format_wangshui_linesX&* 33H= ff% ff% eV$ eV$33B7 $33D9 $r/N) r3r4r5r6rtr~rrrrr1r/r-rmrmws$#0 ((A 6%r/rmc.eZdZdZdZdZdZdZdZy)TestResponseValidatorzTest ResponseValidator class.cd}tj|}|j||j|dd|j|ddy)z/Test JSON extraction from markdown code blocks.z Here is the analysis: ```json { "think": "This is my thinking", "analysis": "This is the analysis" } ``` thinkzThis is my thinkinganalysiszThis is the analysisNr extract_json_from_textassertIsNotNoner<r+textrss r-$test_extract_json_from_text_markdownz:TestResponseValidator.test_extract_json_from_text_markdownsT #99$? V$ *?@  +-CDr/cd}tj|}|j||j|dd|j|ddy)zTest direct JSON extraction.z;{"think": "Direct thinking", "analysis": "Direct analysis"}rzDirect thinkingrzDirect analysisNrrs r-"test_extract_json_from_text_directz8TestResponseValidator.test_extract_json_from_text_directsPL"99$? V$ *;<  +->?r/cTd}tj|}|j|y)zTest handling of invalid JSON.zThis is not JSON at allN)r r assertIsNoners r-#test_extract_json_from_text_invalidz9TestResponseValidator.test_extract_json_from_text_invalids%("99$? &!r/cd}tj|}|j|dd|j|ddy)z)Test manual field extraction using regex.zJResponse with "think": "Manual thinking" and "analysis": "Manual analysis"rzManual thinkingrzManual analysisN)r extract_fields_manuallyr<rs r-test_extract_fields_manuallyz2TestResponseValidator.test_extract_fields_manuallysB["::4@ *;<  +->?r/cd}tj|}|j|dd|j|ddd}tj|}|j|d||j|ddy) z"Test complete response validation.z?{"think": "Complete thinking", "analysis": "Complete analysis"}rzComplete thinkingliuyao_analysiszComplete analysisz$This is just plain text without JSONr"N)r validate_liuyao_responser<)r+responserss r-&test_validate_liuyao_response_completez  124GH:";;HE  12H= "-r/N) r3r4r5r6rrrrrr1r/r-rrs!'E @" @ .r/rceZdZdZedededdZeddZededdZy) TestAnalyzeLiuYaoz&Test the main analyze_liuyao function.z.ai.utils.liuyao_analysis.prepare_liuyao_promptz*ai.utils.liuyao_analysis.LLMServiceFactoryzai.utils.config.get_ai_configcd|_t}d|j_d|_||j_ddd|_t }t |}|j||j|dd|j|dd |j|d d|j|d d |j|jd y)zTest successful analysis. Test promptz7{"think": "Test thinking", "analysis": "Test analysis"}z test-modelz test-provider)providermodelrz Test thinkingrz Test analysisrlanguagezzh-hans completedN) return_valuerget_completionr get_servicerrrr<r%)r+mock_get_ai_config mock_factorymock_prepare_prompt mock_servicerirss r-test_analyze_liuyao_successz-TestAnalyzeLiuYao.test_analyze_liuyao_successs ,9( { 3l ##0) 0<   -7FQ]*^' ' V$ /:  12OD ,7  +Y7 //=r/cd|_t}t|}|j||j |j dy)z4Test analysis failure when prompt preparation fails.Nerror)rrrrr<r%)r+rrirss r-test_analyze_liuyao_no_promptz/TestAnalyzeLiuYao.test_analyze_liuyao_no_promptsB,0(' &! //9r/cd|_t}d|j_||j_t }t |}|j ||j|jdy)z$Test handling of empty LLM response.rr"rN) rrrrrrrr<r%)r+rrrrirss r-"test_analyze_liuyao_empty_responsez4TestAnalyzeLiuYao.test_analyze_liuyao_empty_responsese,9( { 35 ##00<   -' &! //9r/N)r3r4r5r6rrrrr1r/r-rrs{0 ;< 78 *+>,9=>0 ;<:=: ;< 78 :9= :r/rc(eZdZdZdZdZdZdZy) TestEdgeCasesz#Test edge cases and error handling.cxtd}|jd|td}|jd|y)z*Test handling of extreme hexagram numbers.i?BzUnknown hexagramiN)r r@rrs r-test_extreme_hexagram_numbersz+TestEdgeCases.test_extreme_hexagram_numbers.s8#6* (&1#4( (&1r/cTgd}|D]}t|}|jd|!y)z)Test handling of malformed trigram codes.)1234abc12r"zUnknown trigramN)r r@)r+ test_casescaserss r-test_malformed_trigram_codesz*TestEdgeCases.test_malformed_trigram_codes8s*. D%d+F MM+V 4r/cGdd}|} t|}|j|ty#t$rYywxYw)z,Test LiuYao objects with missing attributes.ceZdZdZy)KTestEdgeCases.test_liuyao_with_missing_attributes..IncompleteLiuYaoc d|_d|_y)Nrr)rrr]s r-r.zTTestEdgeCases.test_liuyao_with_missing_attributes..IncompleteLiuYao.__init__Bsr/N)r3r4r5r.r1r/r-IncompleteLiuYaorAs r/rN)rassertIsInstancelistAttributeError)r+rrirss r-#test_liuyao_with_missing_attributesz1TestEdgeCases.test_liuyao_with_missing_attributes?sG   "# ,V4F  ! !&$ /   s!5 AAcddgddddddd d gd }tj|d }|j|t|j d |y)z/Test DataFormatter with complex nested objects.rrN)level2level2brrd)abrgre)cd)level1 list_of_dictscomplexr)r rqrstrr@)r+ complex_datarss r-(test_data_formatter_with_complex_objectsz6TestEdgeCases.test_data_formatter_with_complex_objectsQsd#G, a a   **<C fc* h'r/N)r3r4r5r6rrrrr1r/r-rr+s-25$(r/rc(eZdZdZdZdZdZdZy)TestOutlierCaseszTest outlier and unusual cases.ctgd}|D]/}t|}t|}|j|t1y)z0Test handling of Unicode and special characters.)u测试中文问题？uTest with émojis 🔮✨uMixed 中英文 questionzSpecial chars: @#$%^&*()r")rNrrrr)r+unicode_questionsrrirss r-test_unicode_question_textz+TestOutlierCases.test_unicode_question_textfs: *H2F,V4F  ! !&$ / *r/c gd}|ddD]>}|ddD]4}t||dddd}t|}|j|t6@y)z*Test all possible line value combinations.)rrrrFinvalidNrgrrcr)r+ line_valuesrrrirss r-#test_all_possible_line_combinationsz4TestOutlierCases.test_all_possible_line_combinationsvsV9 bq/B!"1o#rbUuSXY08%%fd3&"r/cgdddgd}tj|}tj|}|j |y)z(Test handling of array-format responses.)z First thoughtzSecond thoughtz Third thoughtzFirst paragraphzSecond paragraph)rrN)jsondumpsr rr)r+array_response json_responserss r-test_response_with_array_formatz0TestOutlierCases.test_response_with_array_formatsFJ*,>?   >2 ";;MJ V$r/cddz}d|d}tj|}|j||jt |dt |y)z%Test handling of very long responses.zVery long analysis iz({"think": "Long thinking", "analysis": "z"}rN)r rrr<r=)r+ long_text long_responserss r-test_extremely_long_responsesz.TestOutlierCases.test_extremely_long_responsessW)D0 CI;cR ";;MJ V$ V$567YHr/N)r3r4r5r6rrrrr1r/r-rrcs)0 4 %Ir/rc,eZdZdZeddZy)TestIntegrationScenarioszTestIntegrationScenarios.test_prepare_prompt_with_complex_datas1J - )*1Q? 1  a( q)Q'a( q) % *)su  @ A[(. '=K $*62F   ( MM. 7 B A As 8B  BN)r3r4r5r6rrr1r/r-rrsF 9:&8;&8r/rceZdZdZdZy)TestResponseValidationSamples7Test response validation with various sample responses.cgd}|D]}|jt|dkDr|dddzn|5tj|}|j |t |j d||j d||jt|ddkDxs!t|ddkDxst|dk(dddy#1swYxYw) r )z;{"think": "Normal thinking", "analysis": "Normal analysis"}z```json { "think": "Markdown thinking", "analysis": "Markdown analysis" } ```z4{"think": "Broken JSON", "analysis": "Missing quote}z;This is just plain text analysis without any JSON structurezTSome text before {"think": "Mixed thinking", "analysis": "Mixed analysis"} and afterzC{"think": ["Think 1", "Think 2"], "analysis": ["Para 1", "Para 2"]}r"aa{"think": "Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking Very long thinking ", "analysis": "Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis Very long analysis "}2Nz...)rrrr)subTestr=r rrdictr@ assertTrue)r+sample_responses response_textrss r- test_response_validation_samplesz>TestResponseValidationSamples.test_response_validation_sampless >.MSEWZ\E\}Sb'9E'Abop*CCMR%%fd3 /8 gv.012Q6,w(1,, &!+qp.pps BCC N)r3r4r5r6rr1r/r-r r s A/r/r )r6r unittest.mockrrr django.testr django.utilsrrai.utils.liuyao_analysisr r r r r rrrrr8rZrmrrrrrr r1r/r-rs  00 !     0<(<:'B8'BTA%A%H4.H4.n6:6:r5(H5(p0Ix0If*8x*8Z2H2r/