diff options
Diffstat (limited to 'tests/test_main.py')
| -rw-r--r-- | tests/test_main.py | 33 |
1 files changed, 33 insertions, 0 deletions
diff --git a/tests/test_main.py b/tests/test_main.py index 321f565..400962e 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -1,6 +1,9 @@ """Test __main__ functions.""" import unittest +import os + from pywal import __main__ +from pywal.settings import __cache_dir__ class TestMain(unittest.TestCase): @@ -35,12 +38,42 @@ class TestMain(unittest.TestCase): __main__.process_args(args) self.assertTrue(args.q) + def test_alpha(self): + """> Test arg parsing (-a)""" + args = __main__.get_args(["-a", "99"]) + __main__.process_args(args) + self.assertTrue(args.a) + def test_ext_script(self): """> Test arg parsing (-o)""" args = __main__.get_args(["-o", "true"]) __main__.process_args(args) self.assertTrue(args.o) + def test_clean(self): + """> Test arg parsing (-c)""" + args = __main__.get_args(["-c"]) + __main__.process_args(args) + self.assertFalse(os.path.isdir(__cache_dir__ / "schemes")) + + def test_reload(self): + """> Test arg parsing (-r)""" + with self.assertRaises(SystemExit): + args = __main__.get_args(["-r"]) + __main__.process_args(args) + + def test_image(self): + """> Test arg parsing (-i)""" + args = __main__.get_args(["-i", "tests/test_files/test.jpg"]) + __main__.process_args(args) + self.assertTrue(args.i) + + def test_json(self): + """> Test arg parsing (-f)""" + args = __main__.get_args(["-f", "tests/test_files/test_file.json"]) + __main__.process_args(args) + self.assertTrue(args.f) + if __name__ == "__main__": unittest.main() |