TensorFlow.NET/test/TensorFlowNET.Graph.UnitTest/ControlFlowTest/CondTestCases.cs

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

namespace TensorFlowNET.UnitTest.ControlFlowTest
{
    /// <summary>
    /// excerpt of tensorflow/python/framework/ops/control_flow_ops_test.py
    /// </summary>
    [TestClass]
    public class CondTestCases : GraphModeTestBase
    {
        [Ignore("Dependent on UpdateEdge")]
        [TestMethod]
        public void testCondTrue_ConstOnly()
        {
            var graph = tf.Graph().as_default();

            var sess = tf.Session(graph);
            var x = tf.constant(2, name: "x");
            var y = tf.constant(5, name: "y");

            var z = control_flow_ops.cond(tf.less(x, y),
                () => tf.constant(22, name: "t22"),
                () => tf.constant(55, name: "f55"));

            int result = z.eval(sess);
            assertEquals(result, 22);
        }

        [TestMethod]
        public void testCondFalse_ConstOnly()
        {
            var graph = tf.Graph().as_default();

            var sess = tf.Session(graph);
            var x = tf.constant(2, name: "x");
            var y = tf.constant(1, name: "y");

            var z = control_flow_ops.cond(tf.less(x, y),
                () => tf.constant(22, name: "t22"),
                () => tf.constant(11, name: "f11"));

            int result = z.eval(sess);
            assertEquals(result, 11);
        }

        [Ignore("Dependent on UpdateEdge")]
        [TestMethod]
        public void testCondTrue()
        {
            tf.Graph().as_default();

            var x = tf.constant(2, name: "x");
            var y = tf.constant(5, name: "y");

            var z = control_flow_ops.cond(tf.less(x, y),
                () => tf.multiply(x, 17),
                () => tf.add(y, 23));

            var result = evaluate<int>(z);
            assertEquals(result, 34);
        }

        [Ignore("Dependent on UpdateEdge")]
        [TestMethod]
        public void testCondFalse()
        {
            tf.Graph().as_default();

            var x = tf.constant(2);
            var y = tf.constant(1);

            var z = control_flow_ops.cond(tf.less(x, y),
                () => tf.multiply(x, 17),
                () => tf.add(y, 23));

            var result = evaluate<int>(z);
            assertEquals(result, 24);
        }

        // NOTE: all other python test cases of this class are either not needed due to strong typing or test a deprecated api

    }
}