def test_files_from_directory(self):
r"""
Runs the files_from_directory command with various args
"""
files = amt.files_from_directory('.', '*')
assert len(files)
files = amt.files_from_directory('.', re.compile('.'))
assert len(files)
def test_calc_percentile_num(self):
r"""
Sends a test array to the calc percentile function
"""
data_list = list(range(100))
val = amt.calc_percentile_num(50, data_list, last=False)
assert val*100 == 50
val = amt.calc_percentile_num(50, data_list, last=True)
assert val*100 == 51
def test_calc_percentile(self):
r"""
Sends a test array to the calc percentile function
"""
data_list = list(range(100))
val = amt.calc_percentile(99, data_list)
assert val == 99
def test_load_infile_list(self):
r"""
Sends a list of infiles
"""
fname1 = os.path.join(FIXTURE_DIR, 'TEST-FRACTURES', 'parallel-plate-01vox.txt')
fname2 = os.path.join(FIXTURE_DIR, 'TEST-FRACTURES', 'parallel-plate-10vox.txt')
infile_list = [fname1, fname2]
#
fields = amt.load_infile_list(infile_list)
assert fields
def test_get_data_vect(self):
r"""
Tests extraction of a vector from a data array
"""
data = sp.arange(100)
data = data.reshape(10, 10)
#
vect = amt.get_data_vect(data, 'x', 0)
assert sp.all(vect == data[0, :])
vect = amt.get_data_vect(data, 'x', 11)
assert sp.all(vect == data[9, :])
vect = amt.get_data_vect(data, 'z', 0)
assert sp.all(vect == data[:, 0])
vect = amt.get_data_vect(data, 'z', 11)
assert sp.all(vect == data[:, 9])
#
with pytest.raises(ValueError):
amt.get_data_vect(data, 'y')
