import unittest

from wrpylib.lib_sledrun_wikitext_from_json import strip_eol


class TestTrimEol(unittest.TestCase):
    def test_trim_eol(self):
        self.assertEqual('', strip_eol(''))
        self.assertEqual(' abc\n', strip_eol(' abc '))
        self.assertEqual('  abc\n  def\ngh\n', strip_eol('  abc\n  def   \ngh '))
        self.assertEqual('', strip_eol('\n\n'))
        self.assertEqual('abc\n', strip_eol('abc     \n'))
        self.assertEqual('\na\n', strip_eol('\na\n\n'))