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

namespace TensorFlowNET.UnitTest.FunctionalOpsTest
{
    /// <summary>
    /// https://www.tensorflow.org/api_docs/python/tf/scan
    /// </summary>
    [TestClass]
    public class ScanTestCase : GraphModeTestBase
    {
        [TestMethod, Ignore("need UpdateEdge API")]
        public void ScanForward()
        {
            var fn = new Func<Tensor, Tensor, Tensor>((a, x) => tf.add(a, x));

            var sess = tf.Session().as_default();

            var input = tf.placeholder(TF_DataType.TF_INT32, new Shape(6));
            var scan = functional_ops.scan(fn, input);
            var result = sess.run(scan, (input, np.array(1, 2, 3, 4, 5, 6)));
            Assert.AreEqual(result, np.array(1, 3, 6, 10, 15, 21));
        }

        [TestMethod, Ignore("need UpdateEdge API")]
        public void ScanReverse()
        {
            var fn = new Func<Tensor, Tensor, Tensor>((a, x) => tf.add(a, x));

            var sess = tf.Session().as_default();

            var input = tf.placeholder(TF_DataType.TF_INT32, new Shape(6));
            var scan = functional_ops.scan(fn, input, reverse: true);
            var result = sess.run(scan, (input, np.array(1, 2, 3, 4, 5, 6)));
            Assert.AreEqual(result, np.array(21, 20, 18, 15, 11, 6));
        }
    }
}