adds test for -e

1 files changed, 8 insertions, 1 deletions

diff --git a/tests/test_main.py b/tests/test_main.py
index 4dbe580..6cf518f 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -1,7 +1,8 @@
 """Test __main__ functions."""
 import unittest
+from unittest.mock import MagicMock
 
-from pywal import __main__
+from pywal import __main__, reload
 from pywal.settings import CACHE_DIR
 
 
@@ -14,6 +15,12 @@ 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)
+        reload.env.assert_not_called()
 
 if __name__ == "__main__":
     unittest.main()