def test_get_maxminabs_index(self):
#def get_max_index(data, axis=1):
"""
Gets the maximum values of a 2D array along an axis
>>> data = [
[4.0, 2.2, 3.0, 5.0, 2.2] # subcase 1
[4.1, 2.1, 3.1, 5.1, 2.1], # subcase 2
>>> max_values, index = get_min_index(data, axis=1)
>>> out = [4.1, 2.2, 3.1, 5.1, 2.2]
>>> index = [1, 0, 1, 1, 0]
"""
data = np.array([
[4.0, 2.2, 3.0, 5.0, 2.2], # subcase 1
[4.1, 2.1, 3.1, 5.1, 2.1],
])
min_values, min_index = get_min_index(data, axis=1)
max_values, max_index = get_max_index(data, axis=1)
abs_values, abs_index = get_abs_index(data, axis=1)
#print(min_index, min_values)
#print(max_index, max_values)
assert np.array_equal(min_index, [0, 1, 0, 0, 1]), min_index
assert np.array_equal(max_index, [1, 0, 1, 1, 0]), max_index
assert np.array_equal(abs_index, [1, 0, 1, 1, 0]), abs_index
assert np.array_equal(min_values, [4.0, 2.1, 3.0, 5.0, 2.1]), min_values
assert np.array_equal(max_values, [4.1, 2.2, 3.1, 5.1, 2.2]), max_values
assert np.array_equal(abs_values, [4.1, 2.2, 3.1, 5.1, 2.2]), abs_values
def test_get_maxminabs_index(self):
# def get_max_index(data, axis=1):
"""
Gets the maximum values of a 2D matrix along an axis
>>> data = [
[4.0, 2.2, 3.0, 5.0, 2.2] # subcase 1
[4.1, 2.1, 3.1, 5.1, 2.1], # subcase 2
>>> max_values, index = get_min_index(data, axis=1)
>>> out = [4.1, 2.2, 3.1, 5.1, 2.2]
>>> index = [1, 0, 1, 1, 0]
"""
data = np.array([[4.0, 2.2, 3.0, 5.0, 2.2], [4.1, 2.1, 3.1, 5.1, 2.1]]) # subcase 1
min_values, min_index = get_min_index(data, axis=1)
max_values, max_index = get_max_index(data, axis=1)
abs_values, abs_index = get_abs_index(data, axis=1)
# print(min_index, min_values)
# print(max_index, max_values)
assert np.array_equal(min_index, [0, 1, 0, 0, 1]), min_index
assert np.array_equal(max_index, [1, 0, 1, 1, 0]), max_index
assert np.array_equal(abs_index, [1, 0, 1, 1, 0]), abs_index
assert np.array_equal(min_values, [4.0, 2.1, 3.0, 5.0, 2.1]), min_values
assert np.array_equal(max_values, [4.1, 2.2, 3.1, 5.1, 2.2]), max_values
assert np.array_equal(abs_values, [4.1, 2.2, 3.1, 5.1, 2.2]), abs_values