def test_tagging(self):
tagger = LiteralNER(NEW_ENTITIES,
[self.tmp_file1.name, self.tmp_file2.name])
s = "Chase notes she's negative for HIV and Hepatitis C"
result = tagger.tag(s.split())
tags = [tag for _, tag in result]
expected_tags = 'O O O O O DISEASE O DISEASE DISEASE'.split()
self.assertEqual(tags, expected_tags)
s = (
"Cuddy points out that the CT scan showed the patient has a metal "
"pin in her arm and can't undergo an MRI")
result = tagger.tag(s.split())
tags = [tag for _, tag in result]
expected_tags = 'O O O O O MEDICAL_TEST MEDICAL_TEST O O O O O O O O O O O O O O MEDICAL_TEST'.split(
)
self.assertEqual(tags, expected_tags)
s = "CT scan said HIV MRI Hepatitis C"
result = tagger.tag(s.split())
tags = [tag for _, tag in result]
expected_tags = 'MEDICAL_TEST MEDICAL_TEST O DISEASE MEDICAL_TEST DISEASE DISEASE'.split(
)
self.assertEqual(tags, expected_tags)
def test_tagging(self):
tagger = LiteralNER(NEW_ENTITIES,
[self.tmp_file1.name, self.tmp_file2.name])
s = "Chase notes she's negative for HIV and Hepatitis C"
result = tagger.tag(s.split())
tags = [tag for _, tag in result]
expected_tags = 'O O O O O DISEASE O DISEASE DISEASE'.split()
self.assertEqual(tags, expected_tags)
s = ("Cuddy points out that the CT scan showed the patient has a metal "
"pin in her arm and can't undergo an MRI")
result = tagger.tag(s.split())
tags = [tag for _, tag in result]
expected_tags = 'O O O O O MEDICAL_TEST MEDICAL_TEST O O O O O O O O O O O O O O MEDICAL_TEST'.split()
self.assertEqual(tags, expected_tags)
s = "CT scan said HIV MRI Hepatitis C"
result = tagger.tag(s.split())
tags = [tag for _, tag in result]
expected_tags = 'MEDICAL_TEST MEDICAL_TEST O DISEASE MEDICAL_TEST DISEASE DISEASE'.split()
self.assertEqual(tags, expected_tags)
