B
    @\	                 @   s   d dl mZmZmZmZ d dlZe d dlmZ d dlm	Z	 d dl
mZ dZdd ZdZd	Zd
ZdZe	eeZdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd ZdS )    )absolute_importdivisionprint_functionunicode_literalsN)assert_allclose)
verify_all)bouncecosinecountforcelinearrepeatsinec                s    fdd}|S )Nc                s     |  d S )N)append)val)results 7lib/python3.7/site-packages/bokeh/tests/test_driving.pyfoo-   s    z_collector.<locals>.foor   )r   r   r   )r   r   
_collector,   s    r   g333333?   g?   c           	   C   sV   g } t ddddgt| }xtdD ]
}|  q(W | ddddddddgksRtd S )Nr            )bdr   r   rangeAssertionError)r   funcir   r   r   test_bounce<   s
    
r!   c              C   sJ   g } t ttttt| }xtdD ]
}|  q&W t| ddddg d S )N   g@g]@g^-@g@)	r   r	   wAphioffsetr   r   r   )r   r   r    r   r   r   test_cosineC   s    
r'   c              C   sD   g } t  t| }xtdD ]
}|  qW | ttdks@td S )Nr   )r   r
   r   r   listr   )r   r   r    r   r   r   
test_countM   s
    
r)   c              C   sh   g } dd dD }t t| |}|  | dgks6t|  | ddgksLt|  | dddgksdtd S )Nc             s   s   | ]
}|V  qd S )Nr   ).0xr   r   r   	<genexpr>V   s    ztest_force.<locals>.<genexpr>)r   barbazr   r-   r.   )r   r   r   r   )r   seqr#   r   r   r   
test_forceT   s    r0   c              C   sH   g } t jdddt| }xtdD ]
}|  q$W t| ddddg d S )Ng      @g@)mbr"   g@gffffff!@gffffff&@)r   r   r   r   r   )r   r   r    r   r   r   test_linear_   s
    
r3   c           	   C   sV   g } t ddddgt| }xtdD ]
}|  q(W | ddddddddgksRtd S )Nr   r   r   r   r   )r   r   r   r   r   )r   r   r    r   r   r   test_repeatf   s
    
r4   c              C   sJ   g } t ttttt| }xtdD ]
}|  q&W t| ddddg d S )Nr"   gc`e@gFpX	@gJv@gd@)	r   r   r#   r$   r%   r&   r   r   r   )r   r   r    r   r   r   	test_sinem   s    
r5   c              C   s~   g } t | }t|}t| | dgks,tt| | ddgksDtt| | dddgks^tt| | ddddgksztd S )Nr   r   r   r   )r   r   Z_advancenextr   )r   Ztestfsr   r   r   test__advance   s    
r8   )Z
__future__r   r   r   r   ZpytestZnumpy.testingr   Zbokeh._testing.util.apir   Zbokeh.drivingZdrivingr   ZALLr   r#   r$   r%   r&   ZTest___all__r!   r'   r)   r0   r3   r4   r5   r8   r   r   r   r   <module>   s(    		

