Use the correct sourcing for all added unit tests

diff --git a/test/test_arnied_wrapper.py b/test/test_arnied_wrapper.py
index f0f5c99..e51349c 100755 (executable)
--- a/test/test_arnied_wrapper.py
+++ b/test/test_arnied_wrapper.py
@@ -29,7 +29,7 @@ import unittest
 import unittest.mock as mock
 import subprocess
 
-import arnied_wrapper
+from src import arnied_wrapper
 
 
 class DummyCmdOutputMapping(object):

diff --git a/test/test_mail_utils.py b/test/test_mail_utils.py
index 772ff7e..7127780 100644 (file)
--- a/test/test_mail_utils.py
+++ b/test/test_mail_utils.py
@@ -23,7 +23,7 @@
 import unittest
 import os
 
-import mail_utils
+from src import mail_utils
 
 
 class MailUtilsTest(unittest.TestCase):