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

namespace Tensorflow.Keras.UnitTest
{
    [TestClass]
    public class OutputTest
    {
        [TestMethod]
        public void OutputRedirectTest()
        {
            using var newOutput = new System.IO.StringWriter();
            tf_output_redirect = newOutput;
            var model = keras.Sequential();
            model.add(keras.Input(shape: 16));
            model.summary();
            string output = newOutput.ToString();
            Assert.IsTrue(output.StartsWith("Model: sequential"));
            tf_output_redirect = null; // don't forget to change it to null !!!!
        }

        [TestMethod]
        public void SwitchOutputsTest()
        {
            using var newOutput = new System.IO.StringWriter();
            var model = keras.Sequential();
            model.add(keras.Input(shape: 16));
            model.summary(); // Console.Out

            tf_output_redirect = newOutput; // change to the custom one
            model.summary();
            string firstOutput = newOutput.ToString();
            Assert.IsTrue(firstOutput.StartsWith("Model: sequential"));

            // if tf_output_reditect is StringWriter, calling "set" will make the writer clear.
            tf_output_redirect = null; // null means Console.Out
            model.summary();

            tf_output_redirect = newOutput; // again, to test whether the newOutput is clear.
            model.summary();
            string secondOutput = newOutput.ToString();
            Assert.IsTrue(secondOutput.StartsWith("Model: sequential"));

            Assert.IsTrue(firstOutput == secondOutput);
            tf_output_redirect = null; // don't forget to change it to null !!!!
        }
    }
}