def setUp(self):
"""Setup files and base data for utility testing."""
self.data_path = example_data
self.data_path_full = example_data_full
data = [
[np.nan, 48.1, 48.3, 65.1, 47.1, 49.9, 49.5, 48.9, 35.5, 44.5],
[51.9, 44.3, 44.5, 58.7, 41.1, 43.9, 43.9, 44.5, 31.1, 40.5],
[48.5, 38.5, 37.1, 53.9, 27.1, 40.7, 34.5, 41.7, 28.7, 37.7],
[33.9, 21.5, 20.1, 48.7, 12.1, 38.5, 13.9, 33.7, 25.3, 28.1],
[19.1, 10.9, 9.9, 42.3, 4.3, 36.5, 6.1, 16.9, 13.9, 13.3],
[12.7, 4.7, 3.9, 31.3, 3.1, 29.1, 3.7, 4.9, 6.9, 4.7],
]
keys = ["V%s" % (i + 1) for i in range(6)]
self.expected_dict = {key: data[i] for i, key in enumerate(keys)}
self.expected_dict_nh = {i: row for i, row in enumerate(data)}
self.expected_arr = np.array(data).T
self.expected_arr_no_nan = deepcopy(self.expected_arr)
self.expected_arr_no_nan[0, 0] = 56.5
self.expected_var_names = keys
self.const_data = deepcopy(self.expected_arr)
self.const_data[:, 0] = np.ones_like(self.expected_arr[:, 0])
self.data, self.var_names = import_data(self.expected_dict)
def test_import_data(self):
"""Test the generalized import data function"""
# File test
data, var_names = utilities.import_data(self.data_path)
self.assertEqual(var_names, self.expected_var_names)
assert_array_equal(data, self.expected_arr)
# dict test
data_dict = utilities.csv_to_dict(self.data_path, dtype=float)
data_arr, var_names = utilities.import_data(data_dict)
self.assertEqual(var_names, self.expected_var_names)
assert_array_equal(data_arr, self.expected_arr)
# array test
data_arr_2, var_names = utilities.import_data(data_arr)
self.assertEqual(var_names, list(range(len(data_arr[0, :]))))
assert_array_equal(data_arr_2, self.expected_arr)
with self.assertRaises(NotImplementedError):
utilities.import_data([0, 1])
def __init__(
self,
data=None,
var_names=None,
x_axis=None,
avg_len=5,
del_const_vars=False,
):
"""Class used to calculated various statistics"""
DVHAStatsBaseClass.__init__(self)
data = TEST_DATA_PATH if data is None else data
self.data, self.var_names = import_data(data, var_names)
self.x_axis = x_axis
self.deleted_vars = []
self.box_cox_data = None
self.avg_len = avg_len
if del_const_vars:
self.del_const_vars()
def test_invalid_data_import(self):
with self.assertRaises(NotImplementedError):
import_data("test")
def test_csv_import(self):
"""Test csv importing"""
data, var_names = import_data(self.data_path)
assert_array_equal(data, self.expected_arr)
self.assertEqual(var_names, self.expected_var_names)
def test_dict_import(self):
"""Test dict importing"""
data, var_names = import_data(self.expected_dict)
assert_array_equal(data, self.expected_arr)
self.assertEqual(var_names, self.expected_var_names)