Merge pull request #73 from aeikenberry/master

Adds test for -e

1 files changed, 9 insertions, 0 deletions

diff --git a/tests/test_main.py b/tests/test_main.py
index 4dbe580..9262c8f 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -1,7 +1,9 @@
 """Test __main__ functions."""
 import unittest
+from unittest.mock import MagicMock
 
 from pywal import __main__
+from pywal import reload
 from pywal.settings import CACHE_DIR
 
 
@@ -14,6 +16,13 @@ class TestMain(unittest.TestCase):
         __main__.process_args(args)
         self.assertFalse((CACHE_DIR / "schemes").is_dir())
 
+    def test_args_e(self):
+        """> Test arg parsing (-e)"""
+        reload.env = MagicMock()
+        args = __main__.get_args(["-e"])
+        __main__.process_args(args)
+        self.assertFalse(reload.env.called)
+
 
 if __name__ == "__main__":
     unittest.main()