def test_all_returned_values_are_equal_to_given_constant(self):
x = 3.5
src = sources.Constant(x)
y = src(size=15)
np.testing.assert_allclose(y, x)
def setUp(self):
self.args = (2, 2)
self.dummy_source = sources.Constant(1)
self.arma = make_random_arma(*self.args, rng=np.random.default_rng(0))
self.arma_alt = make_random_arma(*self.args,
rng=np.random.default_rng(0),
default_source=self.dummy_source)
def setUp(self):
self.n_signals = 3
self.n_samples = 100
self.alphas = [0.99, 0.95]
self.armas = [Arma([_], []) for _ in self.alphas]
self.y0 = 1.0
self.dataset = RandomArmaDataset(
self.n_signals,
self.n_samples,
armas=self.armas,
source=sources.Constant(0),
initial_conditions=([self.y0], []),
)
def setUp(self):
self.x = 3.2
self.src = sources.Constant(self.x)
def test_returns_requested_number_of_values(self):
src = sources.Constant(0.5)
n = 23
y = src(size=n)
self.assertEqual(len(y), n)