ó
ù`]c           @@ sO   d  d l  m Z m Z d  d l Z d  d l m Z d e j f d „  ƒ  YZ d S(   i    (   t   absolute_importt   unicode_literalsN(   t   AnnotationTaskt   TestDisagreementc           B@ s2   e  Z d  Z d „  Z d „  Z d „  Z d „  Z RS(   uN   
    Class containing unit tests for nltk.metrics.agreement.Disagreement.
    c         C@ s>   d d d d d d g } t  | ƒ } |  j | j ƒ  d	 ƒ d
 S(   u}   
        Simple test, based on
        https://github.com/foolswood/krippendorffs_alpha/raw/master/krippendorff.pdf.
        u   coder1u   dress1u   YESu   coder2u   NOu   coder3u   dress2u   dress3g‡¼Š1UUÕ¿N(   u   coder1u   dress1u   YES(   u   coder2u   dress1u   NO(   u   coder3u   dress1u   NO(   u   coder1u   dress2u   YES(   u   coder2u   dress2u   NO(   u   coder3u   dress3u   NO(   R   t   assertAlmostEqualt   alpha(   t   selft   datat   annotation_task(    (    s?   lib/python2.7/site-packages/nltk/test/unit/test_disagreement.pyt	   test_easy   s    	c         C@ s;   d
 d d d d g } t  | ƒ } |  j | j ƒ  d ƒ d	 S(   uœ   
        Same simple test with 1 rating removed.
        Removal of that rating should not matter: K-Apha ignores items with
        only 1 rating.
        u   coder1u   dress1u   YESu   coder2u   NOu   coder3u   dress2g‡¼Š1UUÕ¿N(   u   coder1u   dress1u   YES(   u   coder2u   dress1u   NO(   u   coder3u   dress1u   NO(   u   coder1u   dress2u   YES(   u   coder2u   dress2u   NO(   R   R   R   (   R   R   R   (    (    s?   lib/python2.7/site-packages/nltk/test/unit/test_disagreement.pyt
   test_easy2   s    	c      +   C@ s§   d d d d d d d d d d d d d d  d! d" d# d$ d% d& d' d( d) d* d+ d, d- d. d/ d0 d1 d2 d3 d4 d5 d6 d7 d8 d9 d: d; g) } t  | ƒ } |  j | j ƒ  d ƒ d S(<   ux   
        More advanced test, based on 
        http://www.agreestat.com/research_papers/onkrippendorffalpha.pdf
        u   Au   1u   Bu   Du   2u   Cu   3u   4u   5u   6u   7u   8u   9u   10u   11u   12grË†òÊç?N(   u   Au   1u   1(   u   Bu   1u   1(   u   Du   1u   1(   u   Au   2u   2(   u   Bu   2u   2(   u   Cu   2u   3(   u   Du   2u   2(   u   Au   3u   3(   u   Bu   3u   3(   u   Cu   3u   3(   u   Du   3u   3(   u   Au   4u   3(   u   Bu   4u   3(   u   Cu   4u   3(   u   Du   4u   3(   u   Au   5u   2(   u   Bu   5u   2(   u   Cu   5u   2(   u   Du   5u   2(   u   Au   6u   1(   u   Bu   6u   2(   u   Cu   6u   3(   u   Du   6u   4(   u   Au   7u   4(   u   Bu   7u   4(   u   Cu   7u   4(   u   Du   7u   4(   u   Au   8u   1(   u   Bu   8u   1(   u   Cu   8u   2(   u   Du   8u   1(   u   Au   9u   2(   u   Bu   9u   2(   u   Cu   9u   2(   u   Du   9u   2(   u   Bu   10u   5(   u   Cu   10u   5(   u   Du   10u   5(   u   Cu   11u   1(   u   Du   11u   1(   u   Cu   12u   3(   R   R   R   (   R   R   R   (    (    s?   lib/python2.7/site-packages/nltk/test/unit/test_disagreement.pyt   test_advanced+   sV    	c      +   C@ s§   d d d d d d d d d d d d d d  d! d" d# d$ d% d& d' d( d) d* d+ d, d- d. d/ d0 d1 d2 d3 d4 d5 d6 d7 d8 d9 d: d; g) } t  | ƒ } |  j | j ƒ  d ƒ d S(<   uƒ   
        Same more advanced example, but with 1 rating removed.
        Again, removal of that 1 rating shoudl not matter.
        u   Au   1u   Bu   Du   2u   Cu   3u   4u   5u   6u   7u   8u   9u   10u   11u   12grË†òÊç?N(   u   Au   1u   1(   u   Bu   1u   1(   u   Du   1u   1(   u   Au   2u   2(   u   Bu   2u   2(   u   Cu   2u   3(   u   Du   2u   2(   u   Au   3u   3(   u   Bu   3u   3(   u   Cu   3u   3(   u   Du   3u   3(   u   Au   4u   3(   u   Bu   4u   3(   u   Cu   4u   3(   u   Du   4u   3(   u   Au   5u   2(   u   Bu   5u   2(   u   Cu   5u   2(   u   Du   5u   2(   u   Au   6u   1(   u   Bu   6u   2(   u   Cu   6u   3(   u   Du   6u   4(   u   Au   7u   4(   u   Bu   7u   4(   u   Cu   7u   4(   u   Du   7u   4(   u   Au   8u   1(   u   Bu   8u   1(   u   Cu   8u   2(   u   Du   8u   1(   u   Au   9u   2(   u   Bu   9u   2(   u   Cu   9u   2(   u   Du   9u   2(   u   Bu   10u   5(   u   Cu   10u   5(   u   Du   10u   5(   u   Cu   11u   1(   u   Du   11u   1(   u   Cu   12u   3(   R   R   R   (   R   R   R   (    (    s?   lib/python2.7/site-packages/nltk/test/unit/test_disagreement.pyt   test_advanced2]   sV    	(   t   __name__t
   __module__t   __doc__R	   R
   R   R   (    (    (    s?   lib/python2.7/site-packages/nltk/test/unit/test_disagreement.pyR      s
   			2(   t
   __future__R    R   t   unittestt   nltk.metrics.agreementR   t   TestCaseR   (    (    (    s?   lib/python2.7/site-packages/nltk/test/unit/test_disagreement.pyt   <module>   s   