Changeset 112:c28f2fc2bb05 for nagslang

Timestamp:

Sep 2, 2013, 12:50:30 PM (7 years ago)

Author:

Stefano Rivera <stefano@…>

Branch:

default

Message:

Test with dump_s and load_s

Location:

nagslang

Files:

• tests/test_yamlish.py (modified) (4 diffs)
• yamlish.py (modified) (1 diff)

nagslang/tests/test_yamlish.py

@@ -1 +1 @@
 from unittest import TestCase, SkipTest
-from StringIO import StringIO
 
 try:
@@ -7 +6 @@
     yaml = None  # pyflakes:ignore
 
-from nagslang.yamlish import load, dump
+from nagslang.yamlish import load_s, dump_s
 
 
 class TestParse(TestCase):
-    def assertParsesAs(self, yaml, expected):
-        f = StringIO(yaml.strip())
-        self.assertEqual(load(f), expected)
+    def assertParsesAs(self, text, expected):
+        self.assertEqual(load_s(text.strip()), expected)
 
     def test_dict_list_1(self):
@@ -32 +30 @@
 class TestRoundTrip(TestCase):
     def roundtrip(self, data):
-        f = StringIO()
-        self.dump(data, f)
-        f.seek(0)
-        print '\n=== Begin ===\n%s\n=== End ===' % f.buf.rstrip()
-        self.assertEqual(self.load(f), data)
+        text = self.dump_s(data)
+        print '\n=== Begin ===\n%s\n=== End ===' % text
+        self.assertEqual(self.load_s(text), data)
 
-    def dump(self, data, f):
-        dump(data, f)
+    def dump_s(self, data):
+        return dump_s(data)
 
-    def load(self, f):
-        return load(f)
+    def load_s(self, text):
+        return load_s(text)
 
     def test_simple_dict(self):
@@ -80 +76 @@
 
 class TestFromPyYAML(TestRoundTrip):
-    def dump(self, data, f):
+    def dump_s(self, data):
         if yaml is None:
             raise SkipTest('yaml module unavailable')
-        yaml.dump(data, f, default_flow_style=False)
+        return yaml.dump(data, default_flow_style=False)
 
 
 class TestToPyYAML(TestRoundTrip):
-    def load(self, f):
+    def load_s(self, text):
         if yaml is None:
             raise SkipTest('yaml module unavailable')
-        return yaml.load(f)
+        return yaml.load(text)

nagslang/yamlish.py

@@ -22 +22 @@
 
 def load_s(yaml):
-    return Parser().parse(yaml)
+    return Parser().parse(yaml.strip())