def test_2_sample_model():
"""
Tests sampling function
"""
model = ARIMAX(formula="y ~ x1 + x2", data=data, ar=2, ma=2)
x = model.fit('BBVI', iterations=100, quiet_progress=True)
sample = model.sample(nsims=100)
assert (sample.shape[0] == 100)
assert (sample.shape[1] == len(data) - 2)
def a_test2_sample_model():
"""
Tests sampling function
"""
model = ARIMAX(formula="y ~ x1 + x2",
data=data,
ar=2,
ma=2,
family=Exponential())
x = model.fit('BBVI', iterations=100)
sample = model.sample(nsims=100)
assert (sample.shape[0] == 100)
assert (sample.shape[1] == len(data) - 2)