New custom filter for templates: to_yaml.

1 files changed, 42 insertions, 0 deletions

diff --git a/tests/test_templating.py b/tests/test_templating.py
new file mode 100644
index 0000000..33889ae
--- /dev/null
+++ b/tests/test_templating.py
@@ -0,0 +1,42 @@
+import decimal
+import unittest
+
+from lesana import templating
+
+
+class testFilters(unittest.TestCase):
+    def setUp(self):
+        pass
+
+    def tearDown(self):
+        pass
+
+    def test_to_yaml(self):
+        res = templating.to_yaml(None)
+        self.assertIsInstance(res, str)
+        self.assertEqual(res, 'null')
+
+        s = "A short string"
+        res = templating.to_yaml(s)
+        self.assertEqual(res, s)
+
+        s = """
+        A long, multiline
+        string
+        with multiple
+        lines
+        """
+        res = templating.to_yaml(s)
+        self.assertIsInstance(res, str)
+        self.assertIn('"', res)
+        self.assertIn('\n', res)
+
+        res = templating.to_yaml(10)
+        self.assertEqual(res, '10')
+
+        res = templating.to_yaml(decimal.Decimal('10.1'))
+        self.assertEqual(res, "'10.1'")
+
+
+if __name__ == '__main__':
+    unittest.main()