PCQRSCANER/venv/Lib/site-packages/nltk/test/unit/test_nombank.py
2019-12-22 21:51:47 +01:00

29 lines
796 B
Python

# -*- coding: utf-8 -*-
"""
Unit tests for nltk.corpus.nombank
"""
from __future__ import unicode_literals
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()))