import unittest

import nltk
from nltk.corpus.reader import pl196x


class TestCorpusViews(unittest.TestCase):
    def test_corpus_reader(self):
        pl196x_dir = nltk.data.find("corpora/pl196x")
        pl = pl196x.Pl196xCorpusReader(
            pl196x_dir, r".*\.xml", textids="textids.txt", cat_file="cats.txt"
        )
        pl.tagged_words(fileids=pl.fileids(), categories="cats.txt")