U
    ZÊ¹jØ  ã                   @   sx   d dl mZ d dl mZ d dlmZ d dlmZmZ d dl	m
Z
 ee dd¡d�d	d
„ ƒZee dd¡d�dd„ ƒZdS )é    )Úgiven)Ú
strategies)Údivisors)Údivisor_sigmaÚtotient)Ú	is_squareé   l    d(	 ©Únc                    st   t ˆ ƒ}t|ƒ}tˆ ƒ|d dkks(t‚dd„ |D ƒ}‡ fdd„|D ƒ}dd„ t||ƒD ƒ}ˆ | t|ƒkspt‚d S )Né   r   c                 S   s   g | ]}t |ƒ‘qS © )r   ©Ú.0Úir   r   úb/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/ntheory/tests/test_hypothesis.pyÚ
<listcomp>   s     z'test_tau_hypothesis.<locals>.<listcomp>c                    s   g | ]}t ˆ | ƒ‘qS r   ©r   r   r	   r   r   r      s     c                 S   s   g | ]\}}|| ‘qS r   r   )r   ÚaÚbr   r   r   r      s     )r   Úlenr   ÚAssertionErrorÚzipÚsum)r
   ÚdivZtau_nZsigmasÚtotientsÚmulr   r	   r   Útest_tau_hypothesis   s    r   c                 C   s:   t | ƒ| kst‚t| ƒ}dd„ |D ƒ}| t|ƒks6t‚d S )Nc                 S   s   g | ]}t |ƒ‘qS r   r   r   r   r   r   r      s     z+test_totient_hypothesis.<locals>.<listcomp>)r   r   r   r   )r
   r   r   r   r   r   Útest_totient_hypothesis   s    r   N)Z
hypothesisr   r   ÚstZsympyr   Z%sympy.functions.combinatorial.numbersr   r   Zsympy.ntheory.primetestr   Zintegersr   r   r   r   r   r   Ú<module>   s   

