1 files changed, 43 insertions, 0 deletions
diff --git a/flang-rt/unittests/Runtime/TypeCode.cpp b/flang-rt/unittests/Runtime/TypeCode.cpp
new file mode 100644
index 000000000000..5e694055ffc2
--- /dev/null
+++ b/flang-rt/unittests/Runtime/TypeCode.cpp
@@ -0,0 +1,43 @@
+//===-- unittests/Runtime/TypeCode.cpp --------------------------*- C++ -*-===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#include "gtest/gtest.h"
+#include "flang-rt/runtime/type-code.h"
+
+using namespace Fortran::runtime;
+using namespace Fortran::common;
+
+TEST(TypeCode, ComplexTypes) {
+  // Test all Complex type kinds to ensure they map correctly
+  struct ComplexTypeMapping {
+    int kind;
+    Fortran::ISO::CFI_type_t expectedType;
+  };
+
+  ComplexTypeMapping mappings[] = {
+      {2, CFI_type_half_float_Complex},
+      {3, CFI_type_bfloat_Complex},
+      {4, CFI_type_float_Complex},
+      {8, CFI_type_double_Complex},
+      {10, CFI_type_extended_double_Complex},
+      {16, CFI_type_float128_Complex},
+  };
+
+  for (const auto &mapping : mappings) {
+    TypeCode tc(TypeCategory::Complex, mapping.kind);
+    EXPECT_EQ(tc.raw(), mapping.expectedType)
+        << "Complex kind " << mapping.kind << " should map to CFI type "
+        << mapping.expectedType;
+    EXPECT_TRUE(tc.IsComplex());
+
+    auto categoryAndKind = tc.GetCategoryAndKind();
+    ASSERT_TRUE(categoryAndKind.has_value());
+    EXPECT_EQ(categoryAndKind->first, TypeCategory::Complex);
+    EXPECT_EQ(categoryAndKind->second, mapping.kind);
+  }
+}