#!/usr/bin/env python

import os
import unittest
from textshaper.decorator import lines

class TestDecorators(unittest.TestCase):
    """test decorator(s) for textshaper"""

    def test_lines(self):
        """test line decorator"""

        @lines
        def upper_case(text):
            return [line.upper() for line in text]

        content = """the quick brown fox
jumped over
the lazy dog"""

        _lines = content.splitlines()

        self.assertEqual(upper_case(content), """THE QUICK BROWN FOX
JUMPED OVER
THE LAZY DOG""")
        self.assertEqual(upper_case(_lines), ["THE QUICK BROWN FOX",
                                              "JUMPED OVER",
                                              "THE LAZY DOG"])

if __name__ == '__main__':
    unittest.main()