def hanoi(disks=7,towers=4,limit=None): path = os.path.join(latplan.__path__[0],"puzzles","-".join(map(str,["hanoi",disks,towers]))+".npz") import latplan.puzzles.hanoi as p p.setup() pres = p.generate_random_configs(disks,towers, limit) np.random.shuffle(pres) sucs = [ random.choice(p.successors(c1,disks,towers)) for c1 in pres ] np.savez_compressed(path,pres=pres,sucs=sucs)
def lightsout(type='digital',size=4,limit=None): path = os.path.join(latplan.__path__[0],"puzzles","-".join(map(str,["lightsout",type,size]))+".npz") import importlib p = importlib.import_module('latplan.puzzles.lightsout_{}'.format(type)) p.setup() pres = p.generate_random_configs(size, limit) np.random.shuffle(pres) sucs = [ random.choice(p.successors(c1)) for c1 in pres ] np.savez_compressed(path,pres=pres,sucs=sucs)