[ET][Test] Scalar overflow test macro (#12043)

This PR was created by the merge bot to help merge the original PR into
the main branch.
ghstack PR number: #12013 by
@manuelcandales
^ Please use this as the source of truth for the PR details, comments,
and reviews
ghstack PR base:
https://github.com/pytorch/executorch/tree/gh/manuelcandales/120/base
ghstack PR head:
https://github.com/pytorch/executorch/tree/gh/manuelcandales/120/head
Merge bot PR base:
https://github.com/pytorch/executorch/tree/gh/manuelcandales/119/orig
Merge bot PR head:
https://github.com/pytorch/executorch/tree/gh/manuelcandales/120/orig
@diff-train-skip-merge

---------

Co-authored-by: Manuel Candales <[email protected]>
Co-authored-by: Manuel Candales <[email protected]>

Author: pytorchbot

kernels/test/ScalarOverflowTestMacros.h

@@ -0,0 +1,38 @@
+/*
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+#pragma once
+
+// Macro to generate scalar overflow test cases for a given test suite.
+// The test suite must have a method called expect_bad_scalar_value_dies
+// that takes a template parameter for ScalarType and a Scalar value.
+#define GENERATE_SCALAR_OVERFLOW_TESTS(TEST_SUITE_NAME)         \
+  TEST_F(TEST_SUITE_NAME, ByteTensorTooLargeScalarDies) {       \
+    /* Cannot be represented by a uint8_t. */                   \
+    expect_bad_scalar_value_dies<ScalarType::Byte>(256);        \
+  }                                                             \
+                                                                \
+  TEST_F(TEST_SUITE_NAME, CharTensorTooSmallScalarDies) {       \
+    /* Cannot be represented by a int8_t. */                    \
+    expect_bad_scalar_value_dies<ScalarType::Char>(-129);       \
+  }                                                             \
+                                                                \
+  TEST_F(TEST_SUITE_NAME, ShortTensorTooLargeScalarDies) {      \
+    /* Cannot be represented by a int16_t. */                   \
+    expect_bad_scalar_value_dies<ScalarType::Short>(32768);     \
+  }                                                             \
+                                                                \
+  TEST_F(TEST_SUITE_NAME, FloatTensorTooSmallScalarDies) {      \
+    /* Cannot be represented by a float. */                     \
+    expect_bad_scalar_value_dies<ScalarType::Float>(-3.41e+38); \
+  }                                                             \
+                                                                \
+  TEST_F(TEST_SUITE_NAME, FloatTensorTooLargeScalarDies) {      \
+    /* Cannot be represented by a float. */                     \
+    expect_bad_scalar_value_dies<ScalarType::Float>(3.41e+38);  \
+  }

kernels/test/op_fill_test.cpp

@@ -7,6 +7,7 @@
  */
 
 #include <executorch/kernels/test/FunctionHeaderWrapper.h> // Declares the operator
+#include <executorch/kernels/test/ScalarOverflowTestMacros.h>
 #include <executorch/kernels/test/TestUtil.h>
 #include <executorch/kernels/test/supported_features.h>
 #include <executorch/runtime/core/exec_aten/exec_aten.h>

kernels/test/targets.bzl

@@ -50,6 +50,7 @@ def define_common_targets():
             ],
             exported_headers = [
                 "BinaryLogicalOpTest.h",
+                "ScalarOverflowTestMacros.h",
                 "UnaryUfuncRealHBBF16ToFloatHBF16Test.h",
             ],
             visibility = [