class NERTests(TestCase): def setUp(self): self.parser = NERScanner() def test_person(self): types = self.parser.scan("Jonathan is in the office") self.assertTrue(PiiTypes.PERSON in types) def test_location(self): types = self.parser.scan("Jonathan is in Bangalore") self.assertTrue(PiiTypes.LOCATION in types)
class NERTests(TestCase): def setUp(self): self.parser = NERScanner() def test_person(self): types = self.parser.scan("Roger is in the office") self.assertTrue(PiiTypes.PERSON in types) def test_location(self): types = self.parser.scan("Jonathan is in Bangalore") self.assertTrue(PiiTypes.LOCATION in types) def test_date(self): types = self.parser.scan("Jan 1 2016 is a new year") self.assertTrue(PiiTypes.BIRTH_DATE in types)