def test_white_and_black_lists(self):
doc_terms = ['this', 'is', 'a', 'new', 'test']
white_list = ['test']
black_list = ['this', 'is', 'a']
mx = Matrix(whitelist=white_list, blacklist=black_list)
mx.add_doc(doc_id='doc1',
doc_class='TestClass',
doc_terms=doc_terms,
do_padding=True,
frequency=True)
returned_terms = mx.vocabulary()
expected_terms = ['test']
self.assertItemsEqual(returned_terms, expected_terms)
def test_white_and_black_lists(self):
doc_terms = ['this', 'is', 'a', 'new', 'test']
white_list = ['test']
black_list = ['this', 'is', 'a']
mx = Matrix(whitelist=white_list, blacklist=black_list)
mx.add_doc( doc_id = 'doc1',
doc_class='TestClass',
doc_terms= doc_terms,
do_padding=True,
frequency=True)
returned_terms = mx.vocabulary()
expected_terms = ['test']
self.assertItemsEqual(returned_terms, expected_terms)
