Jim Paris
9082cc9f44
Adjust test expectations accordingly, since the number of intervals they print out will now be smaller.
39 lines
827 B
Python
39 lines
827 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 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:
|
|
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
|
|
|