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

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);
+      }
+    }
+  }
 }