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_empath_not_presesnt(self):
sys.modules['empath'] = None
if sys.version_info.major == 3:
with self.assertRaisesRegex(
Exception,
"Please install the empath library to use FeatsFromSpacyDocAndEmpath."
):
FeatsFromSpacyDocAndEmpath()
else:
with self.assertRaises(Exception):
FeatsFromSpacyDocAndEmpath()
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)