From 0bd16939f1fb9955a93af3cadf77ac37e30a64da Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Sat, 20 May 2017 09:54:37 -0400 Subject: [PATCH] Add utils tests for splitlines and source_to_unicode. --- test/test_utils.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 test/test_utils.py diff --git a/test/test_utils.py b/test/test_utils.py new file mode 100644 index 0000000..7b3298f --- /dev/null +++ b/test/test_utils.py @@ -0,0 +1,27 @@ +from parso.utils import splitlines, source_to_unicode + + +def test_splitlines_no_keepends(): + assert splitlines('asd\r\n') == ['asd', ''] + assert splitlines('asd\r\n\f') == ['asd', '\f'] + assert splitlines('\fasd\r\n') == ['\fasd', ''] + assert splitlines('') == [''] + assert splitlines('\n') == ['', ''] + + +def test_splitlines_keepends(): + assert splitlines('asd\r\n', keepends=True) == ['asd\r\n', ''] + assert splitlines('asd\r\n\f', keepends=True) == ['asd\r\n', '\f'] + assert splitlines('\fasd\r\n', keepends=True) == ['\fasd\r\n', ''] + assert splitlines('', keepends=True) == [''] + assert splitlines('\n', keepends=True) == ['\n', ''] + + +def test_source_to_unicode_unicode_text(self): + source = ( + b"# vim: fileencoding=utf-8\n" + b"# \xe3\x81\x82\xe3\x81\x84\xe3\x81\x86\xe3\x81\x88\xe3\x81\x8a\n" + ) + actual = source_to_unicode(source) + expected = source.decode('utf-8') + assert actual == expected