use concision_core::ex::sample::TestModel;
use concision_core::{Model, ModelFeatures, StandardModelConfig};
use ndarray::prelude::*;
#[test]
fn test_simple_model() {
let features = ModelFeatures::deep(3, 9, 1, 8);
let input = Array1::linspace(1.0, 9.0, features.input());
let mut config = StandardModelConfig::new()
.with_epochs(1000)
.with_batch_size(32);
config.set_learning_rate(0.01);
config.set_momentum(0.9);
config.set_decay(0.0001);
let model = TestModel::<f64>::new(config, features).init();
let output = model.predict(&input);
assert_eq! { output.dim(), (features.output()) }
}