mirror of
https://github.com/dylanaraps/pywal.git
synced 2024-11-29 11:24:03 +01:00
28 lines
704 B
Python
28 lines
704 B
Python
"""Test __main__ functions."""
|
|
import unittest
|
|
from unittest.mock import MagicMock
|
|
|
|
from pywal import __main__, reload
|
|
from pywal.settings import CACHE_DIR
|
|
|
|
|
|
class TestMain(unittest.TestCase):
|
|
"""Test the gen_colors functions."""
|
|
|
|
def test_clean(self):
|
|
"""> Test arg parsing (-c)"""
|
|
args = __main__.get_args(["-c"])
|
|
__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()
|