U
    e»{j  ã                   @   s4   d dl Zd dlZd dlmZmZ dd„ Zdd„ ZdS )é    N)Úcheck_broadcastÚunbroadcastc                	   C   sl   t ddƒdkst‚t dddƒdks&t‚t t¡� t ddƒ W 5 Q R X t t¡� t dddƒ W 5 Q R X d S )N)é
   é   ©é   )r   r   )é   r   r   r   )r   r   r   r   )r   é   )r   r   r	   r   )r   ÚAssertionErrorÚpytestZraisesÚ
ValueError© r   r   ú^/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/utils/tests/test_shapes.pyÚtest_check_broadcast	   s    r   c                  C   sp   t  dddg¡} t  | d¡}t|ƒ}|jdks2t‚t j || ¡ t  d¡} t  | d¡}t|ƒ}|jdkslt‚d S )Nr   r	   r   )r	   r   r   r   )r   é   )r   r   r   )	ÚnpÚarrayZbroadcast_tor   Úshaper
   ZtestingZassert_equalZones)ÚxÚyÚzr   r   r   Útest_unbroadcast   s    
r   )Únumpyr   r   Zastropy.utils.shapesr   r   r   r   r   r   r   r   Ú<module>   s   
