def test_nonexistent_input_file(self):
# Read in the TAFFmat file under test
input_file_basename = os.path.join(
self.test_taffmat_directory,
'nonexistent_taffmat_file.DAT')
self.assertFalse(
taffmat.read_taffmat(input_file_basename))
def test_writing_dat_file_slice(self):
'''
Write the first 1000 elements of data_array to a new .dat and .hdr
file. Then read the new .dat file and make sure it matches the first
1000 elements of the original data_array
'''
slice_output_base_filename = os.path.join(
self.test_taffmat_directory, 'test_slice_output_taffmat')
# number_of_samples_in_slice = self.data_array.shape[1]
number_of_samples_in_slice = 1000
original_data_array = np.copy(
self.data_array[:, 0:number_of_samples_in_slice])
# Write the TAFFmat data slice
taffmat.write_taffmat_slice(
self.data_array, self.header_data,
slice_output_base_filename, 0, number_of_samples_in_slice-1)
# Read the TAFFmat data slice
slice_data_array, slice_time_vector, slice_header_data = \
taffmat.read_taffmat(slice_output_base_filename)
# Confirm the proper number of samples exist
self.assertEqual(
slice_data_array.shape[1],
number_of_samples_in_slice,
'Number of samples in slice incorrect.')
np.testing.assert_array_equal(
slice_data_array,
original_data_array,
'The sliced data array does not equal the original data array.')
def test_input_file_with_dat_extension(self):
# Read in the TAFFmat file under test
input_file_basename = os.path.join(
self.test_taffmat_directory,
'UTEST001.DAT')
data_array, time_vector, header_data = \
taffmat.read_taffmat(input_file_basename)
np.testing.assert_array_equal(
data_array, self.known_data_array,
'Incorrectly read data_array using filename with .DAT extension')
def setUp(self):
self.test_taffmat_directory = os.path.join(
os.path.dirname(os.path.realpath(__file__)),
'test_taffmat_files')
# Read in the TAFFmat file under test
self.input_base_filename = os.path.join(
self.test_taffmat_directory, 'UTEST001')
self.data_array, self.time_vector, self.header_data = \
taffmat.read_taffmat(self.input_base_filename)
def setUp(self):
self.test_taffmat_directory = os.path.join(
os.path.dirname(os.path.realpath(__file__)),
'test_taffmat_files')
# Read in the TAFFmat file under test
self.input_base_filename = os.path.join(
self.test_taffmat_directory, 'UTEST001')
self.data_array, self.time_vector, self.header_data = \
taffmat.read_taffmat(self.input_base_filename)
# Setup the output_basefilename
self.output_base_filename = os.path.join(
self.test_taffmat_directory, 'test_output_taffmat')
# Write the .dat and .hdr files using taffmat.py
taffmat.write_taffmat(
self.data_array, self.header_data, self.output_base_filename)
def test_writing_different_dataset_filename(self):
new_output_base_filename = 'something_different'
taffmat.write_taffmat(self.data_array,
self.header_data,
new_output_base_filename)
data_array, time_vector, header_data = taffmat.read_taffmat(
new_output_base_filename)
self.assertEqual(header_data['dataset'],
new_output_base_filename.upper())
new_output_dat, new_output_hdr = \
self._get_dat_hdr_filenames_from_base(new_output_base_filename)
try:
os.remove(new_output_dat)
except:
print("Couldn't remove the test dat file.")
try:
os.remove(new_output_hdr)
except:
print("Couldn't remove the test hdr file.")
def setUp(self):
test_taffmat_directory = os.path.join(
os.path.dirname(os.path.realpath(__file__)),
'test_taffmat_files')
# Read in the TAFFmat file under test
input_file_basename = os.path.join(
test_taffmat_directory, 'UTEST001')
self.data_array, self.time_vector, self.header_data = \
taffmat.read_taffmat(input_file_basename)
self.beginning_of_raw_data_array = np.array(
[2959, 6291, 9386, 12121], dtype=np.int16)
# Read in the known data array and provide the answers
# for the header file. By having this setup here, we can
# change to a different test file by simply updating
# this section instead of searching all through the test code
known_data_array_input_file = os.path.join(
test_taffmat_directory, 'utest001_data_array_float64.npy')
self.known_data_array = np.load(known_data_array_input_file)
self.known_header = {}
self.known_header['sampling_frequency_hz'] = 96000
self.known_header['number_of_series'] = 2
self.known_header['slope'] = [8e-05, 0.0002]
