Implement TEAM argument for TEAM_NUMBER([TEAM])

Update existing unit test to test this functionality.

Author: nathanweeks

CMakeLists.txt

@@ -719,7 +719,7 @@ if(opencoarrays_aware_compiler)
     add_caf_test(comp_allocated_2 2 comp_allocated_2)
     add_caf_test(alloc_comp_get_convert_nums 2 alloc_comp_get_convert_nums)
     if(NOT CMAKE_Fortran_COMPILER_VERSION VERSION_LESS 8)
-      add_caf_test(team_number 2 team_number)
+      add_caf_test(team_number 6 team_number)
       add_caf_test(teams_subset 3 teams_subset)
       add_caf_test(get_communicator 3  get_communicator)
       add_caf_test(alloc_comp_multidim_shape 2 alloc_comp_multidim_shape)

src/mpi/mpi_caf.c

@@ -8152,12 +8152,10 @@ PREFIX(get_communicator) (caf_team_t *team)
 int
 PREFIX(team_number) (caf_team_t *team)
 {
-  if (team != NULL) caf_runtime_error("team_number does not yet support "
-                                      "the optional team argument");
-
-  // if (used_teams->prev == NULL) 
-  //   return -1;
-  return used_teams->team_list_elem->team_id;
+  if (team != NULL)
+    return ((caf_teams_list *)team)->team_id;
+  else
+    return used_teams->team_list_elem->team_id; /* current team */
 }
 
 void PREFIX(end_team) (caf_team_t *team __attribute__((unused)))

src/tests/unit/teams/team-number.f90

@@ -36,7 +36,9 @@ program main
 
   integer, parameter :: standard_initial_value=-1
 
-  type(team_type), target :: home
+  type(team_type) :: parent, child
+
+  if (num_images() < 8) error stop "I need at least 8 images to function."
 
   call assert(team_number()==standard_initial_value,"initial team number conforms with Fortran standard before 'change team'")
 
@@ -46,11 +48,17 @@ program main
    !! TODO: uncomment the above assertion after implementing support for team_number's optional argument:
 
   after_change_team: block
-    associate(my_team=>mod(this_image(),2)+1)
+    associate(parent_team_number => 100 + (num_images()-1)/4, child_team_number => 1000 + mod(num_images()-1,4)/2)
       !! Prepare for forming two teams: my_team = 1 for even image numbers in the initial team; 2 for odd image numbers
-      form team(my_team,home)
-      change team(home)
-        call assert(team_number()==my_team,"team number conforms with Fortran standard after 'change team'")
+      form team(parent_team_number,parent)
+      change team(parent)
+        call assert(team_number()==parent_team_number,"team number conforms with Fortran standard after 'change team'")
+        form team (child_team_number, child)
+        change team(child)
+          call assert(team_number()==child_team_number,"team number conforms with Fortran standard after 'change team'")
+          call assert(team_number(child)==child_team_number,"team_number(child) conforms with Fortran standard after 'change team'")
+          call assert(team_number(parent)==parent_team_number,"team_number(parent) conforms with Fortran standard")
+        end team
       end team
       call assert(team_number()==standard_initial_value,"initial team number conforms with Fortran standard")
     end associate