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_entities(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.entities(s.split())
expected_entities = [((5, 6), 'DISEASE'), ((7, 9), 'DISEASE')]
self.assertEqual(result, expected_entities)
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.entities(s.split())
expected_entities = [((5, 7), 'MEDICAL_TEST'), ((21, 22), 'MEDICAL_TEST')]
self.assertEqual(result, expected_entities)
s = "CT scan said HIV MRI Hepatitis C"
result = tagger.entities(s.split())
expected_entities = [((0, 2), 'MEDICAL_TEST'), ((3, 4), 'DISEASE'),
((4, 5), 'MEDICAL_TEST'), ((5, 7), 'DISEASE')]
self.assertEqual(result, expected_entities)
def test_entities(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.entities(s.split())
expected_entities = [((5, 6), 'DISEASE'), ((7, 9), 'DISEASE')]
self.assertEqual(result, expected_entities)
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.entities(s.split())
expected_entities = [((5, 7), 'MEDICAL_TEST'),
((21, 22), 'MEDICAL_TEST')]
self.assertEqual(result, expected_entities)
s = "CT scan said HIV MRI Hepatitis C"
result = tagger.entities(s.split())
expected_entities = [((0, 2), 'MEDICAL_TEST'), ((3, 4), 'DISEASE'),
((4, 5), 'MEDICAL_TEST'), ((5, 7), 'DISEASE')]
self.assertEqual(result, expected_entities)
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)
