Updated tests related to previous changes.

mtik00 · Jan 6, 2018 · 3fea7d1 · 3fea7d1
1 parent fbceaa3
commit 3fea7d1
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/test_exceptions/test_utility_functions.py b/tests/test_exceptions/test_utility_functions.py
@@ -6,11 +6,13 @@
 
 
 def test_clear_screen():
-    (is_win, is_lin) = ('win' in sys.platform, 'lin' in sys.platform)
+    is_win = sys.platform.startswith('win')
+    is_nix = sys.platform.startswith('lin') or sys.platform.startswith('darwin')
+
     climenu.IS_WIN = False
-    climenu.IS_LIN = False
+    climenu.IS_NIX = False
 
     with pytest.raises(NotImplementedError):
         climenu.clear_screen()
 
-    (climenu.IS_WIN, climenu.IS_LIN) = (is_win, is_lin)
+    (climenu.IS_WIN, climenu.IS_NIX) = (is_win, is_nix)