From 95d8034a81d32dc0d46ef4e1a4a47e42e508a173 Mon Sep 17 00:00:00 2001 From: Prodesire Date: Mon, 2 Oct 2017 16:52:27 +0800 Subject: [PATCH] add testcase for several strips of utils --- tests/utils/__init__.py | 0 tests/utils/test_strips.py | 23 +++++++++++++++++++++++ 2 files changed, 23 insertions(+) create mode 100644 tests/utils/__init__.py create mode 100644 tests/utils/test_strips.py diff --git a/tests/utils/__init__.py b/tests/utils/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/utils/test_strips.py b/tests/utils/test_strips.py new file mode 100644 index 0000000..c574490 --- /dev/null +++ b/tests/utils/test_strips.py @@ -0,0 +1,23 @@ +from pylib.utils import strips, lstrips, rstrips + + +def test_lstrips(): + assert lstrips('foobbar', '') == 'foobbar' + assert lstrips('foobar', 'fo') == 'obar' + assert lstrips('foofoobar', 'foo') == 'foobar' + assert lstrips('foobarbaz', ('foo', 'bar')) == 'baz' + assert lstrips('foobarbaz', ('bar', 'foo')) == 'barbaz' + + +def test_rstrips(): + assert rstrips('foobbar', '') == 'foobbar' + assert rstrips('foobbar', 'bar') == 'foob' + assert rstrips('foobarbar', 'bar') == 'foobar' + assert rstrips('fozfoobar', ('bar', 'foo')) == 'foz' + assert rstrips('fozfoobar', ('foo', 'bar')) == 'fozfoo' + + +def test_strips(): + assert strips('foobarfoo', '') == 'foobarfoo' + assert strips('foobarfoo', 'foo') == 'bar' + assert strips('foobarfoo', ('foo', 'bar')) == ''