def test_get_data():
"""Test low-level function for data retrieval."""
dat = Data()
dat.generate_mute_data()
idx_list = [(0, 4), (0, 6)]
current_value = (0, 3)
with pytest.raises(RuntimeError):
dat._get_data(idx_list, current_value)
# Test retrieved data for one/two replications in time (i.e., the current
# value is equal to the last sample)
n = 7
d = Data(np.arange(n + 1), 's', normalise=False)
current_value = (0, n)
dat = d._get_data([(0, 1)], current_value)[0]
assert (dat[0][0] == 1)
assert (dat.shape == (1, 1))
d = Data(np.arange(n + 2), 's', normalise=False)
current_value = (0, n)
dat = d._get_data([(0, 1)], current_value)[0]
assert (dat[0][0] == 1)
assert (dat[1][0] == 2)
assert (dat.shape == (2, 1))
# Test retrieval of realisations of the current value.
n = 7
d = Data(np.arange(n), 's', normalise=False)
current_value = (0, n)
dat = d._get_data([current_value], current_value)[0]
