# -*- coding: utf-8 -*-
from static_model import StaticModel
model = StaticModel()
for i, line in enumerate(open("../sensitive.txt")):
if i > 5000:
break
line = unicode(line, "utf-8").strip()
text = line
#text = model.predict_email(text)
text = model.predict_name(text)
#text = model.predict_name_ann(text)
#text = model.predict_street(text)
#text = model.predict_number(text)
if line != text:
print "\n#### Before ####"
print line.encode("utf-8")
print "\n#### After ####"
print text.encode("utf-8")
print "\n=====\n\n"
def test_name(self): model = StaticModel() #prediction = model.predict_name(u"hejsan david, mvh ( Henrik).") #self.assertEqual(prediction,u"hejsan <NAME>, mvh ( <NAME>).") print model.predict_name(u"Mitt namn är Kalle, men du kan kalla mig SAS.")