def test_process_data(self, data_field_class):
r"""
Testing the results of getting data vectors in multiple directions
"""
fmt = '{:4.2f}'
prof = Profile(data_field_class())
prof.args = {'axis': 'x', 'locations': [0, 50, 100]}
prof._process_data()
assert sp.all(prof.processed_data[fmt.format(0)] == prof.data_map[0, :])
assert sp.all(prof.processed_data[fmt.format(50)] == prof.data_map[5, :])
assert sp.all(prof.processed_data[fmt.format(100)] == prof.data_map[9, :])
#
prof.args = {'axis': 'z', 'locations': [0, 50, 100]}
prof._process_data()
assert sp.all(prof.processed_data[fmt.format(0)] == prof.data_map[:, 0])
assert sp.all(prof.processed_data[fmt.format(50)] == prof.data_map[:, 5])
assert sp.all(prof.processed_data[fmt.format(100)] == prof.data_map[:, 9])
#
prof.args = {'axis': 'y', 'locations': [0, 50, 100]}
prof._process_data()
assert prof.processed_data is None
def test_output_data(self, data_field_class):
r"""
Testing output content generation
"""
fmt = '{:4.2f}'
prof = Profile(data_field_class())
prof.infile = os.path.join(TEMP_DIR, 'test-profile.txt')
prof.args = {'axis': 'x', 'locations': [0, 50, 100]}
#
start_ids = [(int(l/100.0*prof.nz)+1) for l in prof.args['locations']]
prof.loc_ids = {fmt.format(loc): sid for loc, sid in zip(prof.args['locations'], start_ids)}
prof.processed_data = {}
#
prof.processed_data[fmt.format(0)] = prof.data_map[0, :]
prof.processed_data[fmt.format(50)] = prof.data_map[5, :]
prof.processed_data[fmt.format(100)] = prof.data_map[9, :]
#
prof._output_data(delim='\t')
assert prof.outfile_content