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]