def setUp(self): self.mapper = WordMapper()
class WordMapperTest(unittest.TestCase): def setUp(self): self.mapper = WordMapper() def testMatchNoun(self): self.assertTrue(self.mapper.match(['n'], PoS.NOUN)) self.assertFalse(self.mapper.match(['vi'], PoS.NOUN)) self.assertFalse(self.mapper.match(['vi'], PoS.NOUN)) self.assertFalse(self.mapper.match(['n'], PoS.NOUN_VSURU)) self.assertTrue(self.mapper.match(['vs'], PoS.NOUN_VSURU)) #self.assertTrue(self.) def testMatchVerb(self): self.assertTrue(self.mapper.match(['vi'], PoS.VERB)) self.assertFalse(self.mapper.match(['vs'], PoS.VERB)) #self.assertTrue() def testMatchAdjective(self): pass def testSuffix(self): self.assertFalse(self.mapper.match(['suf', 'uk', 'n'], PoS.NOUN_SUFFIX)) self.assertFalse(self.mapper.match(['suf', 'ctr'], PoS.NOUN_SUFFIX)) def testTaAuxillary(self): taWords = [['pref', 'n'], ['ok', 'adj-no', 'pn'], ['n'], ['adj-no', 'n-adv', 'n'], ['arch', 'ctr', 'n-suf', 'n'], ['aux-v']] self.assertEqual(5, self.mapper.selectBestWord(taWords, PoS.VERB_AUX)) def testParticle(self): self.assertEqual(1, self.mapper.selectBestWord([['n'], ['prt']], PoS.PRT_CASE)) self.assertEqual(1, self.mapper.selectBestWord([['suf'], ['conj', 'int']], PoS.PRT_CASE)) self.assertEqual(2, self.mapper.selectBestWord([['n'], ['suf', 'n'], ['adv', 'prt']], PoS.PRT_CASE)) def testConjunction(self): self.assertEqual(1, self.mapper.selectBestWord([['n'], ['adv', 'conj', 'uk']], PoS.CONJ)) def testVerb(self): self.assertEqual(1, self.mapper.selectBestWord([['n'], ['vi', 'v5r', 'uk']], PoS.VERB)) def testNoun(self): self.assertEqual(0, self.mapper.selectBestWord([['n'], ['vi', 'v5r', 'uk']], PoS.NOUN)) def testAdj(self): self.assertEqual(1, self.mapper.selectBestWord([['n'], ['vs', 'adj-no']], PoS.ADJ)) def testPrenounAdj(self): self.assertEqual(1, self.mapper.selectBestWord([['n'], ['adj-pn', 'uk']], PoS.ADJ_PRENOUN)) def testAdverb(self): self.assertEqual(1, self.mapper.selectBestWord([['n'], ['adv', 'uk', 'on-mim']], PoS.ADVERB)) def testInterjection(self): self.assertEqual(1, self.mapper.selectBestWord([['n'], ['int', 'uk', 'on-mim']], PoS.INT)) def testPrefix(self): self.assertEqual(1, self.mapper.selectBestWord([['v5s'], ['n', 'vs']], PoS.PREFIX_ADJ)) def testVerbSuffix(self): self.assertEqual(1, self.mapper.selectBestWord([['v5s'], ['suf', 'v5r']], PoS.VERB_SUFFIX)) def testVerbNonIndependent(self): self.assertEqual(1, self.mapper.selectBestWord([['vs'], ['v5r', 'aux']], PoS.VERB_NONIND)) def testNounSuffix(self): self.assertEqual(1, self.mapper.selectBestWord([['v5s'], ['suf', 'n']], PoS.NOUN_SUFFIX))