4d0bfe9f8ea7d868ce309c96e939ce309b93b1c8,naive_bayes/tests/test_plain_text_email.py,TestPlaintextEmailObject,setUp,#TestPlaintextEmailObject#,11

Before Change

        self.plain_file = "./tests/fixtures/plain.eml"
        self.plaintext = io.open(self.plain_file, "r")
        self.text = self.plaintext.read()
        self.plaintext.seek(0)
        self.plain_email = EmailObject(self.plaintext)

    def test_parse_plain_body(self):
        body = self.CLRF.join(self.text.split(self.CLRF)[1:])

After Change

    def setUp(self):
        self.plain_file = "./tests/fixtures/plain.eml"
        with io.open(self.plain_file, "rb") as plaintext:
            self.text = plaintext.read()
            plaintext.seek(0)
            self.plain_email = EmailObject(plaintext)

    def test_parse_plain_body(self):
        body = self.CLRF.join(self.text.split(self.CLRF)[1:])
        self.assertEqual(self.plain_email.body(), body)

Instances