def check_polyaxonfile(polyaxonfile, params=None, profile=None, queue=None, nocache=None, log=True): if not polyaxonfile: polyaxonfile = PolyaxonFile.check_default_path(path=".") if not polyaxonfile: polyaxonfile = "" polyaxonfile = to_list(polyaxonfile) exists = [os.path.isfile(f) for f in polyaxonfile] parsed_params = None if params: parsed_params = parse_params(params) if not any(exists): Printer.print_error("Polyaxonfile is not present, " "please run {}".format(constants.INIT_COMMAND)) sys.exit(1) try: plx_file = PolyaxonFile(polyaxonfile) plx_file = plx_file.get_op_specification(params=parsed_params, profile=profile, queue=queue, nocache=nocache) if log: Printer.print_success("Polyaxonfile valid") return plx_file except Exception as e: handle_cli_error(e, message="Polyaxonfile is not valid.") sys.exit(1)
def test_polyaxon_found(self): def create_file(path, filename, ext): fpath = "{}/{}.{}".format(path, filename, ext) open(fpath, "w") for filename in DEFAULT_POLYAXON_FILE_NAME: for ext in DEFAULT_POLYAXON_FILE_EXTENSION: path = tempfile.mkdtemp() create_file(path, filename, ext) assert PolyaxonFile.check_default_path(path=path)
def test_default_not_found(self): path = tempfile.mkdtemp() assert PolyaxonFile.check_default_path(path=path) is None