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