diff --git a/testmime.py b/testmime.py index c2ff374..56c041b 100644 --- a/testmime.py +++ b/testmime.py @@ -54,7 +54,7 @@ class MimeTestCase(unittest.TestCase): def setUp(self): self.zf = zipfile.ZipFile('test/virus.zip','r') - self.zf.setpassword('denatured') + self.zf.setpassword(b'denatured') def tearDown(self): self.zf.close() diff --git a/testsample.py b/testsample.py index 8f43c71..d6f0256 100644 --- a/testsample.py +++ b/testsample.py @@ -15,7 +15,7 @@ class BMSMilterTestCase(unittest.TestCase): def setUp(self): self.zf = zipfile.ZipFile('test/virus.zip','r') - self.zf.setpassword('denatured') + self.zf.setpassword(b'denatured') def tearDown(self): self.zf.close()