blob: 6cf518f82543f9edb672c865c1bfe7862d5b78fe (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
"""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)
reload.env.assert_not_called()
if __name__ == "__main__":
unittest.main()
|
