Make basic ian can clone scenario work

1 files changed, 4 insertions, 7 deletions

diff --git a/yarnhelper_tests.py b/yarnhelper_tests.py
index 46a0604..cd1c9a3 100644
--- a/yarnhelper_tests.py
+++ b/yarnhelper_tests.py
@@ -66,13 +66,6 @@ class PersistentVariableTests(unittest.TestCase):
         if os.path.exists(yarnhelper.variables_filename):
             os.remove(yarnhelper.variables_filename)
 
-    def test_raises_error_if_no_such_variable(self):
-        h = yarnhelper.YarnHelper()
-        with self.assertRaises(yarnhelper.Error):
-            h.get_variable('FOO')
-            print
-            print 'variables:', h._variables
-
     def test_sets_variable_persistently(self):
         h = yarnhelper.YarnHelper()
         h.set_variable('FOO', 'bar')
@@ -80,6 +73,10 @@ class PersistentVariableTests(unittest.TestCase):
         h2 = yarnhelper.YarnHelper()
         self.assertEqual(h2.get_variable('FOO'), 'bar')
 
+    def test_get_returns_default_if_variable_not_set(self):
+        h = yarnhelper.YarnHelper()
+        self.assertEqual(h.get_variable('FOO', 'bar'), 'bar')
+
     def test_appends_to_empty_list(self):
         h = yarnhelper.YarnHelper()
         h.append_to_list('foo', 1)