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')