Ë
    ·ðÜjá  ã                   ó  — d dl Z d dlZd dlmZ e j                  j                  d  e ee«      j                  «       j                  d   «      «       d dl
mZmZmZmZmZ  G d„ dej                   «      Zedk(  r ej&                  «        yy)é    N)ÚPathé   )ÚBackendÚ
JobRequestÚPrivacyTierÚModeÚroutec                   ó0   — e Zd Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zy)ÚQualityTestsc                 óš   — t        dt        dh«      t        dh«      d¬«      t        dt        dh«      t        dh«      d¬«      g| _        y )NÚcodexÚtextÚanalysiséP   ©Úquality_scoreÚclaudeéZ   )r   Ú	frozensetÚbackends)Úselfs    ú3/home/aria/aria-v3-dev/python/tests/test_quality.pyÚsetUpzQualityTests.setUp   sQ   € äØÜ˜6˜(Ó#Ü˜:˜,Ó'Ø ô	ô ØÜ˜6˜(Ó#Ü˜:˜,Ó'Ø ô	ð
ˆ�ó    c                 óš   — t        ddt        j                  «      }| j                  t	        || j
                  «      j                  d«       y )NÚq1r   r   )r   r   ÚPUBLICÚassertEqualr	   r   Úbackend©r   Újobs     r   Útest_claude_quality_advantagez*QualityTests.test_claude_quality_advantage   s:   € Ü˜˜z¬;×+=Ñ+=Ó>ˆØ×ÑÜ�#�t—}‘}Ó%×-Ñ-¨xõ	
r   c                 ó  — t        dt        dh«      t        dh«      d¬«      t        dt        dh«      t        dh«      d¬«      g}t        ddt        j                  «      }| j                  t        ||«      j                  d«       y )Nr   r   r   r   r   r   Úq2)r   r   r   r   r   r   r	   r   )r   Úequalr!   s      r   Útest_codex_tie_breakerz#QualityTests.test_codex_tie_breaker$   s~   € äØœ F 8Ó,Ü˜:˜,Ó'°rôô Øœ) V HÓ-Ü˜:˜,Ó'°rôð	
ˆô ˜˜z¬;×+=Ñ+=Ó>ˆØ×Ñœ˜s EÓ*×2Ñ2°GÕ<r   c                 óº   — t        ddt        j                  t        j                  ¬«      }| j                  t        || j                  «      j                  d«       y )NÚq3r   ©Úmoder   )	r   r   r   r   ÚCODEXr   r	   r   r   r    s     r   Útest_manual_codexzQualityTests.test_manual_codex2   sF   € ÜØ�*œk×0Ñ0Ü—‘ô
ˆð 	×ÑÜ�#�t—}‘}Ó%×-Ñ-¨wõ	
r   c                 ó¸   — t        ddt        j                  t        j                  ¬«      }| j                  t        || j                  «      j                  «       y )NÚq4r   r)   )	r   r   ÚCONFIDENTIALr   ÚCLAUDEÚassertFalser	   r   Úallowedr    s     r   Útest_privacy_overrides_manualz*QualityTests.test_privacy_overrides_manual;   sA   € ÜØ�*œk×6Ñ6Ü—‘ô
ˆð 	×Ñœ˜s D§M¡MÓ2×:Ñ:Õ;r   c                 óì   — | j                   d   t        dt        dh«      t        dh«      dd¬«      g}t        ddt        j
                  «      }| j                  t        ||«      j                  d	«       y )
Nr   r   r   r   Fr   )Ú	availabler   Úq5r   )	r   r   r   r   r   r   r   r	   r   )r   Úunavailabler!   s      r   Útest_unavailable_claudez$QualityTests.test_unavailable_claudeB   so   € à�M‰M˜!ÑÜØœ) V HÓ-Ü˜:˜,Ó'ØØ ô	ð
ˆô ˜˜z¬;×+=Ñ+=Ó>ˆØ×ÑÜ�#�{Ó#×+Ñ+¨Wõ	
r   N)	Ú__name__Ú
__module__Ú__qualname__r   r"   r&   r,   r3   r8   © r   r   r   r      s    „ ò
ò 
ò=ò
ò<ó
r   r   Ú__main__)ÚsysÚunittestÚpathlibr   ÚpathÚinsertÚstrÚ__file__ÚresolveÚparentsÚaria_router.corer   r   r   r   r	   ÚTestCaser   r9   Úmainr<   r   r   ú<module>rJ      sv   ðÛ 
Û Ý à ‡�‡��‘3‘t˜H“~×-Ñ-Ó/×7Ñ7¸Ñ:Ó;Ô <÷õ ô
C
�8×$Ñ$ô C
ðL ˆzÒØ€H‡M�M…Oð r   