58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
#!/usr/bin/env python
|
|
|
|
import unittest
|
|
import string
|
|
import random
|
|
import sys
|
|
|
|
from i3pystatus.core import util
|
|
|
|
def get_random_string(length=6, chars=string.printable):
|
|
return ''.join(random.choice(chars) for x in range(length))
|
|
|
|
def lchop_factory(prefix, string):
|
|
def test():
|
|
chopped = util.lchop(string, prefix)
|
|
if string.startswith(prefix):
|
|
assert len(chopped) == len(string) - len(prefix)
|
|
if prefix:
|
|
assert not chopped.startswith(prefix)
|
|
test.description = "lchop_test:prefix={}:string={}".format(prefix, string)
|
|
return test
|
|
|
|
def lchop_test_generator():
|
|
nmin = 5
|
|
nmax = 20
|
|
for n in range(nmin, nmax):
|
|
prefix = get_random_string(n)
|
|
string = get_random_string(2*nmax-n)
|
|
yield lchop_factory(prefix, prefix+string)
|
|
yield lchop_factory(prefix, string)
|
|
yield lchop_factory(string, string)
|
|
yield lchop_factory(string, prefix)
|
|
yield lchop_factory("", string)
|
|
|
|
|
|
def partition_factory(iterable, limit, assrt):
|
|
def test():
|
|
partitions = util.partition(iterable, limit)
|
|
partitions = [sorted(partition) for partition in partitions]
|
|
for item in assrt:
|
|
assert sorted(item) in partitions
|
|
test.description = "partition_test:iterable={}:limit={}:expected={}".format(iterable, limit, assrt)
|
|
return test
|
|
|
|
def partition_test_generator():
|
|
cases = [
|
|
([1, 2, 3, 4], 3, [[1,2], [3], [4]]),
|
|
([2, 1, 3, 4], 3, [[1,2], [3], [4]]),
|
|
([0.33, 0.45, 0.89], 1, [[0.33, 0.45, 0.89]]),
|
|
([], 10, []),
|
|
]
|
|
|
|
for iterable, limit, assrt in cases:
|
|
yield partition_factory(iterable, limit, assrt)
|
|
|
|
suite = unittest.TestLoader().loadTestsFromName(__name__)
|
|
unittest.TextTestRunner(verbosity=2).run(suite)
|