def test_main(self): try: from mock import Mock except: from unittest.mock import Mock feat_getter = FeatsFromSpacyDocAndEmpath( empath_analyze_function=mock_empath_analyze) sys.modules['empath'] = Mock(analyze=mock_empath_analyze) FeatsFromSpacyDocAndEmpath() doc = whitespace_nlp('Hello this is a document.') term_freq = feat_getter.get_feats(doc) self.assertEqual( set(term_freq.items()), set({ 'document': 1, 'hello': 1, 'is': 1, 'this': 1, 'a document': 1, 'hello this': 1, 'is a': 1, 'a': 1, 'this is': 1 }.items())) metadata_freq = feat_getter.get_doc_metadata(doc) self.assertEqual(metadata_freq['ridicule'], 1) self.assertNotIn('empath_fashion', metadata_freq)
def test_main(self): try: from mock import Mock except: from unittest.mock import Mock feat_getter = FeatsFromSpacyDocAndEmpath(empath_analyze_function=mock_empath_analyze) sys.modules['empath'] = Mock(analyze=mock_empath_analyze) FeatsFromSpacyDocAndEmpath() doc = whitespace_nlp('Hello this is a document.') term_freq = feat_getter.get_feats(doc) self.assertEqual(set(term_freq.items()), set({'document': 1, 'hello': 1, 'is': 1, 'this': 1, 'a document': 1, 'hello this': 1, 'is a': 1, 'a': 1, 'this is': 1}.items())) metadata_freq = feat_getter.get_doc_metadata(doc) self.assertEqual(metadata_freq['ridicule'], 1) self.assertNotIn('empath_fashion', metadata_freq)
def get_doc_metadata(self, doc, prefix=''): return FeatsFromSpacyDocAndEmpath.get_doc_metadata(self, doc, prefix=prefix)