using Microsoft.VisualStudio.TestTools.UnitTesting;
using static Tensorflow.Binding;

namespace Tensorflow.Keras.UnitTest.Model
{
    [TestClass]
    public class ModelBuildTest
    {
        [TestMethod]
        public void DenseBuild()
        {
            // two dimensions input with unknown batchsize
            var input = tf.keras.layers.Input((17, 60));
            var dense = tf.keras.layers.Dense(64);
            var output = dense.Apply(input);
            var model = tf.keras.Model(input, output);

            // one dimensions input with unknown batchsize
            var input_2 = tf.keras.layers.Input((60));
            var dense_2 = tf.keras.layers.Dense(64);
            var output_2 = dense.Apply(input_2);
            var model_2 = tf.keras.Model(input_2, output_2);

            // two dimensions input with specified batchsize
            var input_3 = tf.keras.layers.Input((17, 60), 8);
            var dense_3 = tf.keras.layers.Dense(64);
            var output_3 = dense.Apply(input_3);
            var model_3 = tf.keras.Model(input_3, output_3);

            // one dimensions input with specified batchsize
            var input_4 = tf.keras.layers.Input((60), 8);
            var dense_4 = tf.keras.layers.Dense(64);
            var output_4 = dense.Apply(input_4);
            var model_4 = tf.keras.Model(input_4, output_4);
        }
    }
}