def test_rainflow_partial_signals_splitturn():
tsgen = TimeSignalGenerator(
10, {
'number': 50,
'amplitude_median': 1.0,
'amplitude_std_dev': 0.5,
'frequency_median': 4,
'frequency_std_dev': 3,
'offset_median': 0,
'offset_std_dev': 0.4
}, None, None)
signal_tot = tsgen.query(10000)
rfc_tot = RF.RainflowCounterThreePoint().process(signal_tot)
turn_points, _ = RF.get_turns(signal_tot)
turn_points = np.insert(turn_points, 0, 0)
turn_num = turn_points.shape[0]
split_points = [
int(np.ceil(turn_num * x)) for x in [0.0, 0.137, 0.23, 0.42, 1.0]
]
rfc_partial = RF.RainflowCounterThreePoint()
for i in range(len(split_points) - 1):
lower = turn_points[split_points[i]]
upper = 10000 if split_points[
i + 1] == turn_points.shape[0] else turn_points[split_points[i +
1]]
_, tot_turns = RF.get_turns(signal_tot[:upper])
rfc_partial.process(signal_tot[lower:upper])
np.testing.assert_array_almost_equal(rfc_tot.loops_from,
rfc_partial.loops_from)
np.testing.assert_array_almost_equal(rfc_tot.loops_to,
rfc_partial.loops_to)
def test_rainflow_partial_signals_get_turns_splitturn():
tsgen = TimeSignalGenerator(
10, {
'number': 50,
'amplitude_median': 1.0,
'amplitude_std_dev': 0.5,
'frequency_median': 4,
'frequency_std_dev': 3,
'offset_median': 0,
'offset_std_dev': 0.4
}, None, None)
signal_tot = tsgen.query(10000)
turns_tot = RF.AbstractRainflowCounter()._get_new_turns(signal_tot)
turn_points, _ = RF.get_turns(signal_tot)
turn_points = np.insert(turn_points, 0, 0)
turn_num = turn_points.shape[0]
split_points = [
int(np.ceil(turn_num * x)) for x in [0.0, 0.137, 0.23, 0.42, 1.0]
]
rfc_partial = RF.AbstractRainflowCounter()
turns_partial = np.empty(0)
for i in range(len(split_points) - 1):
lower = turn_points[split_points[i]]
upper = 10000 if split_points[
i + 1] == turn_points.shape[0] else turn_points[split_points[i +
1]]
turns_partial = np.concatenate(
(turns_partial,
rfc_partial._get_new_turns(signal_tot[lower:upper])))
print(turns_tot[-5:])
print(turns_partial[-5:])
np.testing.assert_array_equal(turns_tot, turns_partial)
def test_get_turns_flat_signal():
samples = np.array([0.0, 0.0, 0.0, 0.0])
expected_index = []
expected_values = []
index, values = RF.get_turns(samples)
np.testing.assert_array_equal(index, expected_index)
np.testing.assert_array_equal(values, expected_values)
def test_get_turns_leading_and_trailing_dups_no_turns():
samples = np.array([0.0, 0.0, 0.5, 1.0, 1.0])
expected_index = []
expected_values = []
index, values = RF.get_turns(samples)
np.testing.assert_array_equal(index, expected_index)
np.testing.assert_array_equal(values, expected_values)
def test_get_turns_leading_and_trailing_dups():
samples = np.array([0., 0., 1., 2., 2., 1., 0., 0., 1., 2., 2.])
expected_index = [3, 6]
expected_values = [2., 0.]
index, values = RF.get_turns(samples)
np.testing.assert_array_equal(index, expected_index)
np.testing.assert_array_equal(values, expected_values)
def test_rainflow_get_turns_trailing_dups():
samples = np.array([1., 2., 1., 1., 1.])
expected_index = [1]
expected_values = [2.]
index, values = RF.get_turns(samples)
np.testing.assert_array_equal(index, expected_index)
np.testing.assert_array_equal(values, expected_values)
def test_rainflow_get_turns_shifted_index_four_trailing_dups():
samples = np.array([
32., 32.1, 32.9, 33., 33., 33., 33., 33., 32.5, 32., 32., 32.7, 37.2,
40., 35.2, 33., 33., 33., 33., 33.
])
expected_index = [3, 9, 13]
expected_values = [33., 32., 40.]
index, values = RF.get_turns(samples)
np.testing.assert_array_equal(index, expected_index)
np.testing.assert_array_equal(values, expected_values)
def test_rainflow_duplicates_no_peak_down():
samples = np.array([3., 2., 2., 1.])
index, values = RF.get_turns(samples)
assert len(index) == 0
assert len(values) == 0
def test_rainflow_partial_get_turns_consecutive_duplicates():
samples = np.array([1., 1., 0.5, 0.5, 1., 1., 1., -1., -1., 0.5, 1.])
index, values = RF.get_turns(samples)
np.testing.assert_array_equal(values, np.array([0.5, 1., -1.]))
np.testing.assert_array_equal(index, np.array([2, 4, 7]))
def test_rainflow_partial_get_turns_no_turns():
samples = np.array([0., 1.])
index, values = RF.get_turns(samples)
assert len(index) == 0
assert len(values) == 0