Fixing a few tests

2017-09-20 23:26:51 -04:00
parent fc2cd296e4
commit db07149e4a
3 changed files with 92 additions and 71 deletions
--- a/tests/test_page.py
+++ b/tests/test_page.py
@@ -141,7 +141,7 @@ def test_page_cycle_theme(reddit, terminal, config, oauth):

        # Previous - will loop to one of the 256 color themes
        page.controller.trigger(curses.KEY_F2)
-        assert page.term.theme.source == 'installed'
+        assert page.term.theme.source in ('preset', 'installed')

        # Reset
        page.term.set_theme()