def setUp(self):
self.training = [['spam', './tests/fixtures/plain.eml'],
['ham', './tests/fixtures/small.eml'],
['scram', './tests/fixtures/plain.eml']]
self.trainer = SpamTrainer(self.training)
with io.open('./tests/fixtures/plain.eml', 'rb') as eml_file:
self.email = EmailObject(eml_file)
def setUp(self):
self.training = [['spam', './tests/fixtures/plain.eml'],
['ham', './tests/fixtures/small.eml'],
['scram', './tests/fixtures/plain.eml']]
self.trainer = SpamTrainer(self.training)
file = io.open('./tests/fixtures/plain.eml', 'r')
self.email = EmailObject(file)
def label_to_training_data(fold_file):
training_data = []
for line in io.open(fold_file, 'rb'):
label_file = line.rstrip().split(b' ')
training_data.append(label_file)
return SpamTrainer(training_data)
