public void InputLayer_CopyLayerForPredictionModel() { var batchSize = 1; var width = 28; var height = 28; var depth = 3; var sut = new InputLayer(height, width, depth); sut.Initialize(1, 1, 1, batchSize, Initialization.GlorotUniform, new Random(232)); var layers = new List <ILayer>(); sut.CopyLayerForPredictionModel(layers); var actual = (InputLayer)layers.Single(); Assert.AreEqual(sut.Width, actual.Width); Assert.AreEqual(sut.Height, actual.Height); Assert.AreEqual(sut.Depth, actual.Depth); }