moulinette/test/test_text.py

from moulinette.utils.text import search, searchf, prependlines, random_ascii


def test_search():
    assert search("a", "a a a") == ["a", "a", "a"]
    assert search("a", "a a a", count=2) == ["a", "a"]
    assert search("a", "a a a", count=-1) == "a"
    assert not search("a", "c c d")


def test_searchf(test_file):
    assert searchf("bar", str(test_file)) == ["bar"]
    assert not searchf("baz", str(test_file))


def test_prependlines():
    assert prependlines("abc\nedf\nghi", "XXX") == "XXXabc\nXXXedf\nXXXghi"
    assert prependlines("", "XXX") == "XXX"


def test_random_ascii():
    import sys
    if sys.version_info[0] == 3:
        assert isinstance(random_ascii(length=2), str)
    else:
        assert isinstance(random_ascii(length=2), unicode)