Changeset 46939

Show
Ignore:
Timestamp:
07/05/06 15:30:42 (2 years ago)
Author:
gracinet
Message:

Fixed broken test

Files:

Legend:

Unmodified
Added
Removed
Modified
Copied
Moved
  • CPS3/products/CPSBayes/trunk/tests/testCPSBayes.py

    r45712 r46939  
    9898                          ['english']) 
    9999 
    100         bayes_tool.unlearn('le la les du un une je il elle de en', 'french') 
     100        del_words = 'le la les du un une je il elle de en' 
     101        bayes_tool.unlearn(del_words, 'french') 
    101102        got = list(bayes_tool.getCategoryList()) 
    102103        got = [element.items() for element in got] 
    103104        got.sort() 
    104105 
     106        # XXX GR: according to warning in BayesCore.storage.zodb, method 
     107        # delWordFromLanguage, it's normal behavior that the french category 
     108        # hasn't been purged. But it should be empty 
    105109        wanted = [[('description', ''), ('name', 'english'), ('label', 'english')], 
     110                  [('description', ''), ('name', 'french'), ('label', 'french')], 
    106111                  [('description', ''), ('name', 'german'), ('label', 'german')], 
    107112                  [('description', ''), ('name', 'spanish'), ('label', 'spanish')]] 
    108113 
    109114        self.assertEquals(got, wanted) 
     115 
     116        # checking that words got deleted 
     117        # GR: language is another concept than category ('fr' by default) 
     118        all_words = set(bayes_tool._getBackend().listWords(language='fr')) 
     119        self.assertEquals(set(del_words.split()) & all_words, set(('de', 'en'))) 
     120 
     121        # checking that remaining words are associated to the spanish category 
     122        res = bayes_tool.guess('en') 
     123        self.assertEquals([cat for cat, prob in res], 
     124                          ['spanish']) 
     125        res = bayes_tool.guess('de') 
     126        self.assertEquals([cat for cat, prob in res], 
     127                          ['spanish']) 
    110128 
    111129    def test_languages(self):