Update import logic to better work with python 3

albertyw · Jan 26, 2020 · e5e0b1c · e5e0b1c
1 parent 7b01f2f
commit e5e0b1c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/mailgun2/__init__.py b/mailgun2/__init__.py
@@ -1,3 +1,5 @@
+from __future__ import absolute_import
+
 __title__ = 'mailgun2'
 __version__ = '1.1.0'
 __author__ = 'Albert Wang'

diff --git a/tests/mailgun.py b/tests/mailgun.py
@@ -1,9 +1,10 @@
+from __future__ import absolute_import
 import os
 import unittest
 
 from mock import patch
 
-import mailgun2
+from mailgun2 import Mailgun
 
 
 class MailgunTestBase(unittest.TestCase):
@@ -12,7 +13,7 @@ def setUp(self):
         self.public_key = 'test_public_key'
         self.private_key = 'test_private_key'
         self.domain = 'test.domain'
-        self.mailgun = mailgun2.Mailgun(
+        self.mailgun = Mailgun(
             self.domain,
             self.public_key,
             self.private_key,