o
    g’×i  ã                   @   s$   d dl Z d dlZdd„ Zdd„ ZdS )é    Nc                 C   s:   |   ¡ }t|ƒD ]\}}||v r|  S qtd|› dƒ‚)Nú'z' not found in code)Ú
splitlinesÚ	enumerateÚ
ValueError)ÚcodeÚsearchÚlinesÚidxÚline© r   úS/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/test/test_compilation.pyÚget_line_of   s   ÿr   c               	   C   s<  t  d¡} tjd| dd}| d¡dksJ ‚t|dƒd t|d	ƒks$J ‚t  d
¡} tjd| dddddd}| d¡dks>J ‚tjd| dddddd}| d¡dksSJ ‚t  d¡} tjd| dddddd}| d¡dksmJ ‚t  d¡} tjd| dddddd}| d¡dks‡J ‚tjd| dddddd}| d¡dksœJ ‚d S )N)é   é   é   za b cT)ÚgraphÚreshaper   Údefé   Úreturn)éx   z(a (b (c d))) -> a b c dr   r   r   é   )r   ÚaÚbÚcÚdz(a (b (c d))) -> (a b) (c d))r   r   r   r   za b c d -> (a (b (c d))))é   é   z(a b) (c d) -> (a (b (c d)))z(a b) (c d) -> a b c d)ÚnpÚzerosÚeinxÚidÚcountr   )Úxr   r   r   r   Útest_compilation   s"   



r$   )r    Únumpyr   r   r$   r   r   r   r   Ú<module>   s    