test(ops): add JUnit tests for unary math ops Square and Abs (#642)

Adds testUnaryMathOps() to GeneratedOperationsTest to verify that the Square and Abs operations correctly execute through the Java API and produce expected scalar TFloat32 results via the native TF engine.

Author: Rohan13253

tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/GeneratedOperationsTest.java

@@ -25,6 +25,7 @@
 import org.tensorflow.ndarray.Shape;
 import org.tensorflow.op.Op;
 import org.tensorflow.op.Ops;
+import org.tensorflow.types.TFloat32;
 import org.tensorflow.types.TInt32;
 
 public final class GeneratedOperationsTest {
@@ -80,4 +81,30 @@ public void testControlDependencies() {
       }
     }
   }
+
+  /**
+   * Test for basic unary math operations. Ensures that the JNI bridge correctly handles Square and
+   * Absolute value functions.
+   */
+  @Test
+  public void testUnaryMathOps() {
+    try (Graph g = new Graph();
+        Session sess = new Session(g)) {
+      Ops ops = Ops.create(g);
+
+      Operand<TFloat32> square = ops.math.square(ops.constant(4.0f));
+      @SuppressWarnings("unchecked")
+      TFloat32 squareRaw = (TFloat32) sess.runner().fetch(square).run().get(0);
+      try (TFloat32 result = squareRaw) {
+        assertEquals(16.0f, result.getFloat(), 0.0f);
+      }
+
+      Operand<TFloat32> abs = ops.math.abs(ops.constant(-5.5f));
+      @SuppressWarnings("unchecked")
+      TFloat32 absRaw = (TFloat32) sess.runner().fetch(abs).run().get(0);
+      try (TFloat32 result = absRaw) {
+        assertEquals(5.5f, result.getFloat(), 0.0f);
+      }
+    }
+  }
 }