def test_init_input_file():
# since we are reading from a file, we expect it to be the same values
# as the file. Since the file matches exp_levels_post, the test checks
# against that list, rather than exp_levels_init even though it is only
# the init
iv = ivdb.IvDb(levels=exp_levelfile)
assert (iv.levels == exp_levels_post)
def test_get_isovol():
"""test get_isovol"""
r = np.full(3, False)
db = test_dir + "/test-isovol/"
iv = ivdb.IvDb(levels=levels, data=data, db=db)
if isdir(db):
shutil.rmtree(db)
mkdir(db)
mkdir(db + '/vols/')
# launch VisIt
import visit
try:
visit.LaunchNowin()
except:
pass
visit.OpenDatabase(test_mesh)
visit.AddPlot('Pseudocolor', data)
# run __get_isovol
lbound = 5
ubound = 15
i = 1
export_res, ubound_out = iv._IvDb__get_isovol(lbound, ubound, i)
# close VisIt
visit.CloseComputeEngine()
# check returned values
if export_res == 1:
r[0] = True
if ubound_out == ubound:
r[1] = True
# check that vol file produced are the same
gen_vol = db + "/vols/1.stl"
exp_vol = exp_vols_dir + "/1.stl"
r[2] = filecmp.cmp(gen_vol, exp_vol)
shutil.rmtree(db)
assert (all(r))
def test_generate_vols_single():
"""Generate all isovolume files with single volume"""
# assert flags
r = np.full(2, False)
# test database path
db = test_dir + "/test-gen-vols-single"
if isdir(db):
shutil.rmtree(db)
# init ivdb obj
iv = ivdb.IvDb(levels=[25], data=data, db=db)
iv.generate_vols(test_mesh)
gen_vols_dir = db + "/vols"
# expected files
exp_db = test_dir + "/exp-single/"
exp_vols_dir = exp_db + "/vols"
common_files = [
f for f in listdir(exp_vols_dir) if isfile(join(exp_vols_dir, f))
]
res = filecmp.cmpfiles(exp_vols_dir, gen_vols_dir, common_files)
match_list = res[0]
non_match = res[1]
# check that files produced are the same
if match_list == common_files:
r[0] = True
if non_match == []:
r[1] = True
# remove files
shutil.rmtree(iv.db)
# check results
assert (all(r))
def __ivdb_obj(completed):
# manually generated a usuable ivdb object
iv = ivdb.IvDb(levels=levels, data=data, db=exp_db)
iv.xmin = iv.ymin = iv.zmin = exp_ext_min
iv.xmax = iv.ymax = iv.zmax = exp_ext_max
iv.completed = completed
return iv
def test_make_db_dir():
db = test_dir + "/test-mkdir"
if isdir(db):
shutil.rmtree(db)
iv = ivdb.IvDb(levels=levels, data=data, db=db)
iv._IvDb__make_db_dir()
res = False
if isdir(db):
res = True
shutil.rmtree(iv.db)
assert (res)
def test_init_input():
r = np.full(4, False)
iv = ivdb.IvDb(levels=levels, data=data, db=exp_db)
if iv.levels == exp_levels_init:
r[0] = True
if iv.data == data:
r[1] = True
if iv.db == exp_db:
r[2] = True
if iv.completed is False:
r[3] = True
assert (all(r))
def test_init_none():
r = np.full(4, False)
iv = ivdb.IvDb()
if iv.levels is None:
r[0] = True
if iv.data is None:
r[1] = True
if iv.db == getcwd() + "/tmp":
r[2] = True
if iv.completed is False:
r[3] = True
assert (all(r))
def test_check_data_nodata():
"""check levels, no levels"""
r = np.full(2, False)
iv = ivdb.IvDb(levels=[-5, 0, 45], data=data)
# no levels remaining, creates error
with pytest.raises(RuntimeError) as error_info:
iv._IvDb__check_data(test_mesh)
if "No data exists" in str(error_info):
r[0] = True
if iv.levels == []:
r[1] = True
assert (all(r))
def test_check_data_outofbounds():
"""test check levels, data out of bounds"""
r = np.full(4, False)
# level -5 and 45 are out of bounds, so two warnings expected
iv = ivdb.IvDb(levels=[-5, 5, 15, 25, 35, 45], data=data)
# data out of range should produce warning
with warnings.catch_warnings(record=True) as w:
iv._IvDb__check_data(test_mesh)
warnings.simplefilter("always")
exp_str = "out of data bounds"
r[0:3] = __check_warning(w, [exp_str] * 2, 2)
if iv.levels == exp_levels_init:
# arbmax has not yet been added to levels list so should match
# init levels
r[3] = True
assert (all(r))
def test_write_levels():
r = np.full(2, False)
db = test_dir + "/test-write-levels/"
# this test goes right to the write step and skips that addition
# of the arbmax value in the process, so we will init with a level
# list that already includes the arbmax value and expect it to match
# the level file that would be present after the complete process
iv = ivdb.IvDb(levels=exp_levels_post, data=data, db=db)
if isdir(db):
shutil.rmtree(db)
mkdir(db)
iv.write_levels()
levelfile_out = db + "/levelfile"
if isfile(levelfile_out):
r[0] = True
r[1] = __compare_levelfiles(levelfile_out, exp_levelfile)
shutil.rmtree(db)
assert (all(r))
def main():
# get args
args = parse_arguments()
# generate level info if necessary
# levels is either a list of values or path to file
levels = get_levels(args)
# get database information
db = os.getcwd() + '/' + args.db[0]
# get dataname
data = args.dataname[0]
# run steps depending on mode
mode = args.which
visit_modes = ["full", "visit"]
moab_modes = ["full", "moab"]
iv = None # initialize
if mode in visit_modes:
iv = ivdb.IvDb(levels=levels, data=data, db=db)
driver.generate_volumes(iv, args.meshfile[0])
if mode in moab_modes:
if args.tags:
tags = process_tags(args.tags)
else:
tags = None
# pass IvDb info if object exists from previous step
if iv is not None:
ig = isg.IsGm(ivdb=iv)
else:
ig = isg.IsGm(levels=levels, data=data, db=db)
# create geom from info
driver.create_geometry(ig,
tag_for_viz=args.tagviz,
norm=args.norm[0],
tags=tags,
sname=args.geomfile[0],
sdir=args.savepath[0])
def test_make_db_dir_exists():
r = np.full(4, False)
db = test_dir + "/test-direxists"
if isdir(db):
shutil.rmtree(db)
mkdir(db)
db_exp = test_dir + "/test-direxists_1/"
iv = ivdb.IvDb(levels=levels, data=data, db=db)
with warnings.catch_warnings(record=True) as w:
iv._IvDb__make_db_dir()
warnings.simplefilter("always")
r[0:2] = __check_warning(w, ["exists"], 1)
if iv.db == db_exp:
r[2] = True
if isdir(iv.db):
r[3] = True
shutil.rmtree(iv.db)
shutil.rmtree(db)
assert (all(r))
def test_get_isovol_nodata():
"""test get_isovol no data present"""
r = np.full(6, False)
db = test_dir + "/test-isovol-nodata/"
iv = ivdb.IvDb(levels=[5, 15, 25, 28, 35], data=data, db=db)
if isdir(db):
shutil.rmtree(db)
mkdir(db)
mkdir(db + '/vols/')
# launch VisIt
import visit
try:
visit.LaunchNowin()
except:
pass
visit.OpenDatabase(test_mesh)
visit.AddPlot('Pseudocolor', data)
# run __get_isovol
lbound = 25
ubound = 28
i = 3
with warnings.catch_warnings(record=True) as w:
export_res, ubound_out = iv._IvDb__get_isovol(lbound, ubound, i)
# close VisIt
visit.CloseComputeEngine()
# check returned/changed values
if export_res == 0:
r[0] = True
if ubound_out == 35:
r[1] = True
if iv.levels == [5, 15, 25, 35]:
r[2] = True
gen_vol = db + "/vols/3.stl"
if not isfile(gen_vol):
# no file should be generated
r[3] = True
r[4:6] = __check_warning(w, ["no data to export between"], 1)
shutil.rmtree(db)
assert (all(r))
def test_check_data():
"""test check levels, all data good"""
r = np.full(5, False)
iv = ivdb.IvDb(levels=levels, data=data)
arbmin, arbmax, minext, maxext = iv._IvDb__check_data(test_mesh)
exp_min = -10
exp_max = 50
exp_minext = [-10., -10., -10.]
exp_maxext = [10., 10., 10.]
if arbmin == exp_min:
r[0] = True
if arbmax == exp_max:
r[1] = True
if iv.levels == exp_levels_init:
# arbmax has not yet been added to levels list so should match
# init levels
r[2] = True
if list(minext) == exp_minext:
r[3] = True
if list(maxext) == exp_maxext:
r[4] = True
assert (all(r))
def test_generate_vols():
"""Generate all isovolume files."""
# assert flags
r = np.full(2, False)
# test database path
db = test_dir + "/test-gen-vols"
if isdir(db):
shutil.rmtree(db)
# init ivdb obj
iv = ivdb.IvDb(levels=levels, data=data, db=db)
iv.generate_vols(test_mesh)
# check that files produced are the same
gen_vols_dir = db + "/vols"
res = filecmp.cmpfiles(exp_vols_dir, gen_vols_dir, common_files)
match_list = res[0]
non_match = res[1]
if match_list == common_files:
r[0] = True
if non_match == []:
r[1] = True
# remove files
shutil.rmtree(iv.db)
# check results
assert (all(r))