def test_number_of_saves_in_forward(nt, ncp):
cp = SimpleCheckpoint()
f = SimpleOperator()
b = SimpleOperator()
rev = Revolver(cp, f, b, ncp, nt)
assert(cp.save_counter == 0)
rev.apply_forward()
assert(cp.save_counter == min(ncp, nt-1))
def test_forward_nt(nt, ncp):
cp = SimpleCheckpoint()
f = SimpleOperator()
b = SimpleOperator()
rev = Revolver(cp, f, b, ncp, nt)
assert(f.counter == 0)
rev.apply_forward()
assert(f.counter == nt)
def test_ptr_loads_and_saves(nt, ncp):
cp = SimpleCheckpoint()
f = SimpleOperator()
b = SimpleOperator()
rev = Revolver(cp, f, b, ncp, nt)
rev.apply_forward()
rev.apply_reverse()
assert(cp.save_pointers == cp.load_pointers)
assert(len(cp.save_pointers) == min(ncp, nt - 1))
def test_num_loads_and_saves(nt, ncp):
cp = SimpleCheckpoint()
f = SimpleOperator()
b = SimpleOperator()
rev = Revolver(cp, f, b, ncp, nt)
rev.apply_forward()
assert(cp.load_counter == 0)
rev.apply_reverse()
assert(cp.load_counter >= cp.save_counter)
def test_num_loads_and_saves(nt, ncp, singlefile, diskckp):
cp = SimpleCheckpoint()
f = SimpleOperator()
b = SimpleOperator()
rev = SingleLevelRevolver(cp, f, b, ncp, nt,
diskstorage=diskckp,
filedir="./",
singlefile=singlefile)
rev.apply_forward()
assert(cp.load_counter == 0)
rev.apply_reverse()
assert(cp.load_counter >= cp.save_counter)
def test_num_loads_and_saves(nt, mwd, mrd, dwd, drd, uf, ub, singlefile):
cp = SimpleCheckpoint()
f = SimpleOperator()
b = SimpleOperator()
npStorage = NumpyStorage(cp.size, nt, cp.dtype, wd=mwd, rd=mrd)
dkStorage = DiskStorage(cp.size,
nt,
cp.dtype,
filedir="./",
singlefile=singlefile,
wd=dwd,
rd=drd)
st_list = [npStorage, dkStorage]
rev = MultiLevelRevolver(cp, f, b, nt, storage_list=st_list, uf=uf, ub=ub)
rev.apply_forward()
assert cp.load_counter == 0
rev.apply_reverse()
assert cp.load_counter >= cp.save_counter