[flang] Added tests for intrinsic function 'team_number()'
reviewers: klausler, PeteSteinfeld
Differential Revision: https://reviews.llvm.org/D100897
GitOrigin-RevId: 9dd3965e96d1f31f0d9303d503bbd12129cfe92b
diff --git a/docs/Intrinsics.md b/docs/Intrinsics.md
index f8f335f..d5fdee8 100644
--- a/docs/Intrinsics.md
+++ b/docs/Intrinsics.md
@@ -746,7 +746,7 @@
| Intrinsic Category | Intrinsic Procedures Lacking Support |
| --- | --- |
-| Coarray intrinsic functions | LCOBOUND, UCOBOUND, FAILED_IMAGES, GET_TEAM, IMAGE_INDEX, STOPPED_IMAGES, TEAM_NUMBER, COSHAPE |
+| Coarray intrinsic functions | LCOBOUND, UCOBOUND, FAILED_IMAGES, IMAGE_INDEX, STOPPED_IMAGES, COSHAPE |
| Object characteristic inquiry functions | ALLOCATED, ASSOCIATED, EXTENDS_TYPE_OF, IS_CONTIGUOUS, PRESENT, RANK, SAME_TYPE, STORAGE_SIZE |
| Type inquiry intrinsic functions | BIT_SIZE, DIGITS, EPSILON, HUGE, KIND, MAXEXPONENT, MINEXPONENT, NEW_LINE, PRECISION, RADIX, RANGE, TINY|
| Non-standard intrinsic functions | AND, OR, XOR, LSHIFT, RSHIFT, SHIFT, ZEXT, IZEXT, COSD, SIND, TAND, ACOSD, ASIND, ATAND, ATAN2D, COMPL, DCMPLX, EQV, NEQV, INT8, JINT, JNINT, KNINT, LOC, QCMPLX, DREAL, DFLOAT, QEXT, QFLOAT, QREAL, DNUM, NUM, JNUM, KNUM, QNUM, RNUM, RAN, RANF, ILEN, SIZEOF, MCLOCK, SECNDS, COTAN, IBCHNG, ISHA, ISHC, ISHL, IXOR, IARG, IARGC, NARGS, NUMARG, BADDRESS, IADDR, CACHESIZE, EOF, FP_CLASS, INT_PTR_KIND, ISNAN, MALLOC |
diff --git a/test/Semantics/team_number.f90 b/test/Semantics/team_number.f90
new file mode 100644
index 0000000..fc489cc
--- /dev/null
+++ b/test/Semantics/team_number.f90
@@ -0,0 +1,22 @@
+! RUN: %python %S/test_errors.py %s %flang_fc1
+! Check for semantic errors in team_number() function calls
+
+subroutine test
+ use, intrinsic :: iso_fortran_env, only: team_type
+ type(team_type) :: team
+
+ ! correct calls, should produce no errors
+ team = get_team()
+ print *, team_number()
+ print *, team_number(team)
+ print *, team_number(team=team)
+
+ ! call with too many arguments
+ !ERROR: too many actual arguments for intrinsic 'team_number'
+ print *, team_number(1, 3)
+
+ ! keyword argument with incorrect type
+ !ERROR: Actual argument for 'team=' has bad type 'REAL(4)'
+ print *, team_number(team=3.1415)
+
+end subroutine
diff --git a/unittests/Evaluate/intrinsics.cpp b/unittests/Evaluate/intrinsics.cpp
index dd1a8a5..45e47af 100644
--- a/unittests/Evaluate/intrinsics.cpp
+++ b/unittests/Evaluate/intrinsics.cpp
@@ -294,6 +294,30 @@
.Push(Const(Scalar<Real4>{}))
.DoCall(); // bad type
+ TestCall{defaults, table, "team_number"}.DoCall(Int4::GetType());
+ TestCall{defaults, table, "team_number"}
+ .Push(Const(Scalar<Int4>{}))
+ .Push(Const(Scalar<Int4>{}))
+ .DoCall(); // too many args
+ TestCall{defaults, table, "team_number"}
+ .Push(Named("bad", Const(Scalar<Int4>{})))
+ .DoCall(); // bad keyword
+ TestCall{defaults, table, "team_number"}
+ .Push(Const(Scalar<Int4>{}))
+ .DoCall(); // bad type
+ TestCall{defaults, table, "team_number"}
+ .Push(Const(Scalar<Char>{}))
+ .DoCall(); // bad type
+ TestCall{defaults, table, "team_number"}
+ .Push(Const(Scalar<Log4>{}))
+ .DoCall(); // bad type
+ TestCall{defaults, table, "team_number"}
+ .Push(Const(Scalar<Complex8>{}))
+ .DoCall(); // bad type
+ TestCall{defaults, table, "team_number"}
+ .Push(Const(Scalar<Real4>{}))
+ .DoCall(); // bad type
+
// TODO: test other intrinsics
// Test unrestricted specific to generic name mapping (table 16.2).