def load_naadsm(): arguments = docopt.docopt(__doc__, version="farmspread") if arguments["-v"]: logging.basicConfig(level=logging.DEBUG) elif arguments["-q"]: logging.basicConfig(level=logging.ERROR) else: logging.basicConfig(level=logging.INFO) metadata = dict() if arguments["--runs"]: runs = int(arguments["--runs"]) assert (runs > 0) else: runs = 1 if arguments["--stream"]: print(arguments) stream = int(arguments["--stream"]) assert (stream > 0) else: stream = 1 if arguments["--out"]: outfile = arguments["--out"] outfile = re.sub("^\~", os.environ["HOME"], outfile) else: outfile = "out.h5" if arguments["--chunk"]: chunk_size = int(arguments["--chunk"]) else: chunk_size = runs no_direct = arguments["--nodirect"] dataformat.clear_datafile(outfile) scfile = util.check_filename(arguments["--scenario"], "scenario file") hfile = util.check_filename(arguments["--herd"], "herd file") metadata.update({ "run_cnt": runs, "stream_idx": stream, "chunk_size": chunk_size, "naadsm_args": str(arguments), "herd": hfile, "scenario": scfile, "uuid": str(uuid.uuid4()), "executable": "naadsm.py" }) for chunk_idx, run_cnt in util.ChunkIter(runs, chunk_size): results, stream = multirun(scfile, hfile, no_direct, run_cnt, stream + chunk_size) dataformat.save_runs(outfile, results, metadata)
def load_naadsm(): arguments = docopt.docopt(__doc__, version="farmspread") if arguments["-v"]: logging.basicConfig(level=logging.DEBUG) elif arguments["-q"]: logging.basicConfig(level=logging.ERROR) else: logging.basicConfig(level=logging.INFO) metadata=dict() if arguments["--runs"]: runs=int(arguments["--runs"]) assert(runs>0) else: runs=1 if arguments["--stream"]: print(arguments) stream=int(arguments["--stream"]) assert(stream>0) else: stream=1 if arguments["--out"]: outfile=arguments["--out"] outfile=re.sub("^\~", os.environ["HOME"], outfile) else: outfile="out.h5" if arguments["--chunk"]: chunk_size=int(arguments["--chunk"]) else: chunk_size=runs no_direct=arguments["--nodirect"] dataformat.clear_datafile(outfile) scfile=util.check_filename(arguments["--scenario"], "scenario file") hfile=util.check_filename(arguments["--herd"], "herd file") metadata.update({"run_cnt" : runs, "stream_idx" : stream, "chunk_size" : chunk_size, "naadsm_args" : str(arguments), "herd" : hfile, "scenario" : scfile , "uuid" : str(uuid.uuid4()), "executable" : "naadsm.py" }) for chunk_idx, run_cnt in util.ChunkIter(runs, chunk_size): results, stream=multirun(scfile, hfile, no_direct, run_cnt, stream+chunk_size) dataformat.save_runs(outfile, results, metadata)
def test_nonexist(self): with self.assertRaises(RuntimeError): util.check_filename("nonexistent", "hopefully not existent")
def test_exist(self): f=tempfile.NamedTemporaryFile() util.check_filename(f.name, "this module file.")
def test_exist(self): f = tempfile.NamedTemporaryFile() util.check_filename(f.name, "this module file.")