From 2f3212fe8dbb8bddaa4cca07ccbf1ab738a8b595 Mon Sep 17 00:00:00 2001
From: Plamen Dimitrov <pdimitrov@pevogam.com>
Date: Tue, 23 Oct 2018 15:58:26 +0800
Subject: [PATCH] Use the correct sourcing for all added unit tests

---
 test/test_arnied_wrapper.py |    2 +-
 test/test_mail_utils.py     |    2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/test_arnied_wrapper.py b/test/test_arnied_wrapper.py
index f0f5c99..e51349c 100755
--- 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
--- 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):
-- 
1.7.1