def _test(db: DemoDB, save_inc: int, update_inc: int): # init db.int_val = 1 # save db.int_val += save_inc db.save() assert_load_same(db, 1 + save_inc) def _cb(val: DemoDB) -> Any: val.int_val += update_inc return val.int_val db.update(_cb) assert_load_same(db, 1 + save_inc + update_inc)