o
    ýÑÐb1  ã                   @   sŠ   d Z ddlZddlmZ ddlZddlmZ ddlm	Z	 ddl
mZ ddlmZ G dd„ deƒZG d	d
„ d
ejƒZedkrCe ¡  dS dS )zDTests for the pylint checker in :mod:`pylint.extensions.emptystring
é    N)Úcheckers)ÚCompareToZeroChecker)ÚPyLinter)ÚBaseReporterc                   @   s   e Zd Zdd„ Zdd„ ZdS )ÚCompareToZeroTestReporterc                 C   s   | j  |¡ d S ©N)ÚmessagesÚappend)ÚselfÚmsg© r   úh/var/www/secure340b-portal/env/lib/python3.10/site-packages/pylint/test/extensions/test_comparetozero.pyÚhandle_message   s   z(CompareToZeroTestReporter.handle_messagec                 C   s
   g | _ d S r   )r   )r
   ÚmoduleÚfilepathr   r   r   Úon_set_current_module   s   
z/CompareToZeroTestReporter.on_set_current_moduleN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r      s    r   c                   @   s    e Zd Zedd„ ƒZdd„ ZdS )ÚCompareToZeroUsedTCc                 C   sD   t ƒ | _| j tƒ ¡ t | j¡ | j t| jƒ¡ | j d¡ d S )NÚI)	r   Ú_linterZset_reporterr   r   Z
initializeZregister_checkerr   Údisable)Úclsr   r   r   Ú
setUpClass!   s
   zCompareToZeroUsedTC.setUpClassc                 C   s´   t  t  t  t¡¡dd¡}| j |g¡ | jjj}|  	t
|ƒd¡ |D ]}|  	|jd¡ |  	|jd¡ q#|  	|d jd¡ |  	|d jd	¡ |  	|d
 jd¡ |  	|d jd¡ d S )NÚdatazcompare_to_zero.pyé   zcompare-to-zerozAvoid comparisons to zeror   é   é   é	   é   é   é   é   )ÚospÚjoinÚdirnameÚabspathÚ__file__r   ÚcheckÚreporterr   ÚassertEqualÚlenÚsymbolr   Úline)r
   Z	elif_testZmsgsr   r   r   r   Útest_comparetozero_message)   s   ÿ
z.CompareToZeroUsedTC.test_comparetozero_messageN)r   r   r   Úclassmethodr   r/   r   r   r   r   r      s    
r   Ú__main__)Ú__doc__ÚosÚos.pathÚpathr$   ZunittestZpylintr   Zpylint.extensions.comparetozeror   Zpylint.lintr   Zpylint.reportersr   r   ZTestCaser   r   Úmainr   r   r   r   Ú<module>   s   	ÿ