From ed70cc2eb21c79bdadfdbc2e0b94880d7837d810 Mon Sep 17 00:00:00 2001 From: ZyX Date: Wed, 6 Aug 2014 13:47:11 +0400 Subject: [PATCH] Fix tests --- tests/test_provided_config_files.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/tests/test_provided_config_files.py b/tests/test_provided_config_files.py index a6371588..90dc3d4e 100644 --- a/tests/test_provided_config_files.py +++ b/tests/test_provided_config_files.py @@ -116,9 +116,14 @@ class TestConfig(TestCase): config_overrides = None theme_overrides = {} - with IpyPowerline() as powerline: - segment_info = Args(prompt_count=1) - for prompt_type in ['in', 'in2', 'out', 'rewrite']: + segment_info = Args(prompt_count=1) + + with IpyPowerline(True) as powerline: + for prompt_type in ['in', 'in2']: + powerline.render(matcher_info=prompt_type, segment_info=segment_info) + powerline.render(matcher_info=prompt_type, segment_info=segment_info) + with IpyPowerline(False) as powerline: + for prompt_type in ['out', 'rewrite']: powerline.render(matcher_info=prompt_type, segment_info=segment_info) powerline.render(matcher_info=prompt_type, segment_info=segment_info)