def test_email(self): """ Test parsing content """ for email, content in self._emails.items(): expected = json.loads(read_email_from_file( email.replace(".txt", ".json"))) email_parser = Email() email_parser.parse(content) self.assertEquals(expected['To'], email_parser.to_address) self.assertEquals(expected['From'], email_parser.from_address)
def test_email(self): """ Test parsing content """ for email, content in self._emails.items(): expected = json.loads( read_email_from_file(email.replace(".txt", ".json"))) email_parser = Email() email_parser.parse(content) self.assertEquals(expected['To'], email_parser.to_address) self.assertEquals(expected['From'], email_parser.from_address)
def setUp(self): """ Read in all sample(s) """ self._emails = {} os.chdir(CWD) for filename in glob.glob('raw_email_??.txt'): self._emails[filename] = read_email_from_file(filename)