def main():
# write message
tnam = os.path.splitext(os.path.basename(__file__))[0]
msg = "Running {} test".format(tnam)
print(msg)
# determine if largetest directory exists
dir_available = is_directory_available(example_basedir)
if not dir_available:
return
# get a list of test models to run
example_dirs = get_mf6_models()
# run the test models
for on_dir in example_dirs:
sim = Simulation(
on_dir,
mf6_regression=True,
cmp_verbose=False,
make_comparison=set_make_comparison(on_dir),
)
run_mf6(sim)
return
def test_mf6model():
# determine if largetest directory exists
dir_available = is_directory_available(example_basedir)
if not dir_available:
return
# get a list of test models to run
example_dirs = get_mf6_models()
# run the test models
for on_dir in example_dirs:
yield run_mf6, Simulation(
on_dir,
mf6_regression=True,
cmp_verbose=False,
make_comparison=set_make_comparison(on_dir),
)
return
def get_mf6_models():
"""
Get a list of test models
"""
# determine if test directory exists
dir_avail = is_directory_available(example_basedir)
if not dir_avail:
return []
# determine if running on travis
is_CI = "CI" in os.environ
# get current branch
if is_CI:
branch = os.path.basename(os.environ["GITHUB_REF"])
else:
branch = get_branch()
print("On branch {}".format(branch))
# tuple of example files to exclude
# exclude = (None,)
exclude = ("test205_gwtbuy-henrytidal", )
# update exclude
if is_CI:
exclude_CI = (None, )
exclude = exclude + exclude_CI
exclude = list(exclude)
# write a summary of the files to exclude
print("list of tests to exclude:")
for idx, ex in enumerate(exclude):
print(" {}: {}".format(idx + 1, ex))
# build list of directories with valid example files
if example_basedir is not None:
example_dirs = get_example_dirs(example_basedir,
exclude,
prefix="test")
else:
example_dirs = []
# exclude dev examples on master or release branches
if "master" in branch.lower() or "release" in branch.lower():
drmv = []
for d in example_dirs:
if "_dev" in d.lower():
drmv.append(d)
for d in drmv:
example_dirs.remove(d)
# determine if only a selection of models should be run
select_example_dirs = None
select_packages = None
for idx, arg in enumerate(sys.argv):
if arg.lower() == "--sim":
if len(sys.argv) > idx + 1:
select_example_dirs = sys.argv[idx + 1:]
break
elif arg.lower() == "--pak":
if len(sys.argv) > idx + 1:
select_packages = sys.argv[idx + 1:]
select_packages = [item.upper() for item in select_packages]
break
elif arg.lower() == "--match":
if len(sys.argv) > idx + 1:
like = sys.argv[idx + 1]
example_dirs = [item for item in example_dirs if like in item]
break
# determine if the selection of model is in the test models to evaluate
if select_example_dirs is not None:
example_dirs = get_select_dirs(select_example_dirs, example_dirs)
if len(example_dirs) < 1:
msg = "Selected models not available in test"
print(msg)
# determine if the specified package(s) is in the test models to evaluate
if select_packages is not None:
example_dirs = get_select_packages(select_packages, example_basedir,
example_dirs)
if len(example_dirs) < 1:
msg = "Selected packages not available ["
for pak in select_packages:
msg += " {}".format(pak)
msg += "]"
print(msg)
return example_dirs
def get_mf5to6_models():
"""
Get a list of test models
"""
# determine if test directory exists
dir_available = is_directory_available(example_basedir)
if not dir_available:
return []
# list of example files to exclude
exclude = (None, )
# write a summary of the files to exclude
print("list of tests to exclude:")
for idx, ex in enumerate(exclude):
print(" {}: {}".format(idx + 1, ex))
# build list of directories with valid example files
if example_basedir is not None:
example_dirs = get_example_dirs(
example_basedir,
exclude,
prefix="test",
find_sim=False,
)
else:
example_dirs = []
# determine if only a selection of models should be run
select_dirs = None
select_packages = None
for idx, arg in enumerate(sys.argv):
if arg.lower() == "--sim":
if len(sys.argv) > idx + 1:
select_dirs = sys.argv[idx + 1:]
break
elif arg.lower() == "--pak":
if len(sys.argv) > idx + 1:
select_packages = sys.argv[idx + 1:]
select_packages = [item.upper() for item in select_packages]
break
# determine if the selection of model is in the test models to evaluate
if select_dirs is not None:
example_dirs = get_select_dirs(select_dirs, example_dirs)
if len(example_dirs) < 1:
msg = "Selected models not available in test"
print(msg)
# determine if the specified package(s) is in the test models to evaluate
if select_packages is not None:
example_dirs = get_select_packages(select_packages, example_basedir,
example_dirs)
if len(example_dirs) < 1:
msg = "Selected packages not available ["
for idx, pak in enumerate(select_packages):
msg += "{}".format(pak)
if idx + 1 < len(select_packages):
msg += ", "
msg += "]"
print(msg)
return example_dirs
def get_mf6_models():
"""
Get a list of test models
"""
# determine if examples directory exists
dir_available = is_directory_available(example_basedir)
if not dir_available:
return []
# determine if running on travis
is_CI = "CI" in os.environ
# tuple of example files to exclude
exclude = ("ex-gwf-csub-p02c",)
# update exclude
if is_CI:
exclude_CI = (None,)
exclude = exclude + exclude_CI
exclude = list(exclude)
# write a summary of the files to exclude
print("list of tests to exclude:")
for idx, ex in enumerate(exclude):
print(" {}: {}".format(idx + 1, ex))
# build list of directories with valid example files
if example_basedir is not None:
example_dirs = get_example_dirs(example_basedir, exclude, prefix="ex-")
else:
example_dirs = []
# determine if only a selection of models should be run
select_dirs = None
select_packages = None
for idx, arg in enumerate(sys.argv):
if arg.lower() == "--sim":
if len(sys.argv) > idx + 1:
select_dirs = sys.argv[idx + 1 :]
break
elif arg.lower() == "--pak":
if len(sys.argv) > idx + 1:
select_packages = sys.argv[idx + 1 :]
select_packages = [item.upper() for item in select_packages]
break
# determine if the selection of model is in the test models to evaluate
if select_dirs is not None:
example_dirs = get_select_dirs(select_dirs, example_dirs)
if len(example_dirs) < 1:
msg = "Selected models not available in test"
print(msg)
# determine if the specified package(s) is in the test models to evaluate
if select_packages is not None:
example_dirs = get_select_packages(
select_packages, example_basedir, example_dirs
)
if len(example_dirs) < 1:
msg = "Selected packages not available ["
for pak in select_packages:
msg += " {}".format(pak)
msg += "]"
print(msg)
return example_dirs
