File size: 416 Bytes
a5be142 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from pvnet.models.baseline.single_value import Model
import pytest
@pytest.fixture()
def single_value_model(model_minutes_kwargs):
model = Model(**model_minutes_kwargs)
return model
def test_model_forward(single_value_model, sample_batch):
y = single_value_model(sample_batch)
# check output is the correct shape
# batch size=2, forecast_len=15
assert tuple(y.shape) == (2, 16), y.shape
|