using Microsoft.VisualStudio.TestTools.UnitTesting;
using Tensorflow.Keras.Layers;
using static Tensorflow.Binding;
using static Tensorflow.KerasApi;

namespace Tensorflow.Keras.UnitTest
{
    [TestClass]
    public class InitLayerNameTest
    {
        [TestMethod]
        public void RNNLayerNameTest() 
        {
            var simpleRnnCell = keras.layers.SimpleRNNCell(1);
            Assert.AreEqual("simple_rnn_cell", simpleRnnCell.Name);
            var simpleRnn = keras.layers.SimpleRNN(2);
            Assert.AreEqual("simple_rnn", simpleRnn.Name);
            var lstmCell = keras.layers.LSTMCell(2);
            Assert.AreEqual("lstm_cell", lstmCell.Name);
            var lstm = keras.layers.LSTM(3);
            Assert.AreEqual("lstm", lstm.Name);
        }

        [TestMethod]
        public void ConvLayerNameTest()
        {
            var conv2d = keras.layers.Conv2D(8, activation: "linear");
            Assert.AreEqual("conv2d", conv2d.Name);
            var conv2dTranspose = keras.layers.Conv2DTranspose(8);
            Assert.AreEqual("conv2d_transpose", conv2dTranspose.Name);
        }
    }
}