45 lines
1012 B
Python
45 lines
1012 B
Python
# Just some helpers for test functions
|
|
|
|
import shutil, os
|
|
|
|
def myrepr(x):
|
|
if isinstance(x, basestring):
|
|
return '"' + x + '"'
|
|
else:
|
|
return repr(x)
|
|
|
|
def eq_(a, b):
|
|
if not a == b:
|
|
raise AssertionError("%s != %s" % (myrepr(a), myrepr(b)))
|
|
|
|
def lt_(a, b):
|
|
if not a < b:
|
|
raise AssertionError("%s is not less than %s" % (myrepr(a), myrepr(b)))
|
|
|
|
def in_(a, b):
|
|
if a not in b:
|
|
raise AssertionError("%s not in %s" % (myrepr(a), myrepr(b)))
|
|
|
|
def ne_(a, b):
|
|
if not a != b:
|
|
raise AssertionError("unexpected %s == %s" % (myrepr(a), myrepr(b)))
|
|
|
|
def lines_(a, n):
|
|
l = a.count('\n')
|
|
if not l == n:
|
|
if len(a) > 5000:
|
|
a = a[0:5000] + " ... truncated"
|
|
raise AssertionError("wanted %d lines, got %d in output: '%s'"
|
|
% (n, l, a))
|
|
|
|
def recursive_unlink(path):
|
|
try:
|
|
shutil.rmtree(path)
|
|
except OSError:
|
|
pass
|
|
try:
|
|
os.unlink(path)
|
|
except OSError:
|
|
pass
|
|
|