B
    ×\>  ã               @   sF   d dl Z d dlZd dlZG dd„ dejƒZddd„ZedkrBeƒ  dS )	é    Nc               @   sN   e Zd Zejd dkreefneZdd„ Zdd„ Z	dd„ Z
d	d
„ Zdd„ ZdS )ÚTestr   é   c             C   s   |   tt ¡ | jƒ¡ d S )N)Ú
assertTrueÚ
isinstanceÚmklZget_cpu_clocksÚ	_int_long)Úself© r	   ú'lib/python3.7/site-packages/mkl/test.pyÚtest_cpu_clocks
   s    zTest.test_cpu_clocksc             C   s   |   tt ¡ tƒ¡ d S )N)r   r   r   Zget_cpu_frequencyÚfloat)r   r	   r	   r
   Útest_cpu_frequency   s    zTest.test_cpu_frequencyc             C   s4   |   ttjd¡ |   ttjd¡ |   ttjd¡ d S )NÚ2éÿÿÿÿr   )ZassertRaisesÚ	TypeErrorr   Zset_num_threadsÚ
ValueError)r   r	   r	   r
   Útest_set_num_threads   s    zTest.test_set_num_threadsc             C   s   |   tt ¡ tƒ¡ d S )N)r   r   r   Zget_max_threadsÚint)r   r	   r	   r
   Útest_get_max_threads   s    zTest.test_get_max_threadsc             C   sH   t  ¡ }|  t|tƒ¡ |  t|d | jƒ¡ |  t|d | jƒ¡ d S )Nr   é   )r   Zmem_statr   r   Útupler   )r   Úresultr	   r	   r
   Útest_mem_stat   s    zTest.test_mem_statN)Ú__name__Ú
__module__Ú__qualname__ÚsysÚversion_infor   Zlongr   r   r   r   r   r   r	   r	   r	   r
   r      s   r   r   c             C   sh   t dtj ƒ t dtj ƒ t dtj ƒ t dt ¡  ƒ t ¡ }| 	t 
t¡¡ tj| d}| |¡S )Nzsys.prefix: %szsys.version: %szmkl-service version: %rzMKL version string: %r)Ú	verbosity)Úprintr   ÚprefixÚversionr   Ú__version__Zget_version_stringÚunittestZ	TestSuiteZaddTestZ	makeSuiter   ZTextTestRunnerÚrun)r   ZsuiteZrunnerr	   r	   r
   r$       s    r$   Ú__main__)r   )r   r#   r   ZTestCaser   r$   r   r	   r	   r	   r
   Ú<module>   s   
