You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

29 lines
796 B
Python

5 years ago
# -*- 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()))