Prepare tests for performance optimizations

1 files changed, 12 insertions, 2 deletions

diff --git a/test/segments/vi_mode.spec b/test/segments/vi_mode.spec
index 205c38b7..5bed38bd 100755
--- a/test/segments/vi_mode.spec
+++ b/test/segments/vi_mode.spec
@@ -7,31 +7,41 @@ SHUNIT_PARENT=$0
 
 function setUp() {
   export TERM="xterm-256color"
-  # Load Powerlevel9k
-  source powerlevel9k.zsh-theme
 }
 
 function testViInsertModeWorks() {
   local KEYMAP='viins'
 
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
   assertEquals "%K{black} %F{blue}INSERT " "$(prompt_vi_mode left 1 false)"
 }
 
 function testViInsertModeWorksWhenLabeledAsMain() {
   local KEYMAP='main'
 
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
   assertEquals "%K{black} %F{blue}INSERT " "$(prompt_vi_mode left 1 false)"
 }
 
 function testViCommandModeWorks() {
   local KEYMAP='vicmd'
 
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
   assertEquals "%K{black} %F{white}NORMAL " "$(prompt_vi_mode left 1 false)"
 }
 
 function testViInsertModeStringIsCustomizable() {
   local KEYMAP='viins'
 
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
   assertEquals "%K{black} %F{blue}INSERT " "$(prompt_vi_mode left 1 false)"
 }