def test_begin(self):
plug = Profile()
plug.pfile = tempfile.mkstemp()[1]
try:
plug.begin()
assert plug.prof
finally:
plug.finalize(None)
def test_begin(self):
plug = Profile()
plug.pfile = tempfile.mkstemp()[1]
try:
plug.begin()
assert plug.prof
finally:
plug.finalize(None)
def test_finalize(self):
def func():
pass
plug = Profile()
plug.begin()
plug.prepareTest(func)
pfile = plug.pfile
try:
assert os.path.exists(pfile)
finally:
plug.finalize(None)
assert not os.path.exists(pfile), "finalize did not remove temp file %s" % pfile
def test_finalize(self):
def func():
pass
plug = Profile()
plug.begin()
plug.prepareTest(func)
pfile = plug.pfile
try:
assert os.path.exists(pfile)
finally:
plug.finalize(None)
assert not os.path.exists(pfile), \
"finalize did not remove temp file %s" % pfile
def test_prepare_test(self):
r = {}
class dummy:
def runcall(self, f, r):
r[1] = f(), "wrapped"
def func():
return "func"
plug = Profile()
plug.prof = dummy()
result = plug.prepareTest(func)
try:
result(r)
assert r[1] == ("func", "wrapped")
finally:
plug.finalize(None)
def test_prepare_test(self):
r = {}
class dummy:
def runcall(self, f, r):
r[1] = f(), "wrapped"
def func():
return "func"
plug = Profile()
plug.prof = dummy()
result = plug.prepareTest(func)
try:
result(r)
assert r[1] == ("func", "wrapped")
finally:
plug.finalize(None)
