Spaces:
Sleeping
Sleeping
File size: 760 Bytes
d916065 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
"""
Unit tests for nltk.corpus.nombank
"""
import unittest
from nltk.corpus import nombank
# Load the nombank once.
nombank.nouns()
class NombankDemo(unittest.TestCase):
def test_numbers(self):
# No. of instances.
self.assertEqual(len(nombank.instances()), 114574)
# No. of rolesets
self.assertEqual(len(nombank.rolesets()), 5577)
# No. of nouns.
self.assertEqual(len(nombank.nouns()), 4704)
def test_instance(self):
self.assertEqual(nombank.instances()[0].roleset, "perc-sign.01")
def test_framefiles_fileids(self):
self.assertEqual(len(nombank.fileids()), 4705)
self.assertTrue(all(fileid.endswith(".xml") for fileid in nombank.fileids()))
|