
    i9
              
       P   d Z ddlZddlmc mZ ddlZddlZddl	Z	ddl
mZ g dZdedefdZ ed        ed	        ed       eD ]  \  ZZ ed
dddedd      Z ee      Zej+                  dd      Zej+                  dd      dd Zej+                  di       Z ede d        ede d        ede        edk(  r;er9 edej+                  d               edej+                  dg       dd   d!        ed"e d!         ed#        ed$        ed       y)%zGTest Visual Decider with varied data signals - not hardcoded in prompt.    N)format_prompt))zincreased fivefoldzDThe company's market share increased fivefold between 2010 and 2020.)zwent from X to Yz7Obesity rates went from 13% to 42% in just two decades.)zdropped by halfzBAfter the regulation, emissions dropped by half within five years.)skyrocketedz5Housing prices skyrocketed while wages remained flat.)zbiographical factz(She was born in a small village in 1892.)zsingle number no trendz"The building stands 432 feet tall.promptreturnc           	      
   	 t        j                  dd| ddddgddd	      }t        j                  |j                  j                               }|j                  d
|j                        }t        j                  ddt        |            }t        j                  dd|      }t        j                  d|      }|r#t        j                  |j                               S 	 i S # t        $ r}t        d|        Y d}~i S d}~ww xY w)z$Call Claude and parse JSON response.claudez--printz--modelsonnetz--output-formatjsonTx   )capture_outputtexttimeoutresultz^```json\s* z\s*```$z\{[\s\S]*\}zError: N)
subprocessrunr
   loadsstdoutstripgetresubstrsearchgroup	Exceptionprint)r   r   wrappercontent
json_matches         H/home/sietch6/trending-topics-pipeline/test_visual_decider_variations.pycall_claude_jsonr#      s    y&)X?PRXY	
 **V]]0023++h6&&S\:&&R1YY~w7
::j..011  I  smIs   CC   	D)C==DzF======================================================================z/TESTING VISUAL DECIDER WITH VARIED DATA SIGNALStest   z	Test Beatz,Test documentary about trends and statisticsz(No specific research))
segment_id
time_starttime_end	beat_name	voiceoverpremiseresearch_excerptdecisionERRORthinkingr   d   data_visualizationz
[]z  VO: ""z  DECISION: DATA_VISUALIZATIONz	  CHART: 
chart_typez  DATA: data   z...z  THINKING: zG
======================================================================zFEXPECTED: First 4 should be DATA_VISUALIZATION, last 2 should be IMAGE)__doc__builtins@py_builtins_pytest.assertion.rewrite	assertionrewrite
@pytest_arr
   r   r   'prompts.prompt_v4.layer4_visual_deciderr   
TEST_CASESr   dictr#   r   labelr*   r   r   r   r-   r/   dv     r"   <module>rF      sg   M     	 A	
S T ( h 7 8 h" (E9>1F f%Fzz*g.Hzz*b)$3/H 
("	-B	Cwa.	HYKr
"#	L

#$''B	"&&./01+BQ/045	L
#
&'1(4 o  N O hrE   