stat_dev: add error_stop

Author: jvdp1

src/stdlib_experimental_stat_mean.f90

@@ -1,6 +1,7 @@
 submodule (stdlib_experimental_stat) stdlib_experimental_stat_mean
 
 
+use stdlib_experimental_error, only: error_stop
 use stdlib_experimental_optval, only: optval
 implicit none
 
@@ -38,6 +39,8 @@ module function mean_2_sp_sp(mat, dim) result(res)
 
     dim_ = optval(dim, 1)
 
+    if (dim_ < 0 .or. dim_ > 2 ) call error_stop("ERROR (mean): invalid argument (dim) ")
+
     allocate(res(size(mat, dim_)))
 
     if (dim_ == 1) then
@@ -61,6 +64,8 @@ module function mean_2_dp_dp(mat, dim) result(res)
 
     dim_ = optval(dim, 1)
 
+    if (dim_ < 0 .or. dim_ > 2 ) call error_stop("ERROR (mean): invalid argument (dim) ")
+
     allocate(res(size(mat, dim_)))
 
     if (dim_ == 1) then
@@ -84,6 +89,8 @@ module function mean_2_qp_qp(mat, dim) result(res)
 
     dim_ = optval(dim, 1)
 
+    if (dim_ < 0 .or. dim_ > 2 ) call error_stop("ERROR (mean): invalid argument (dim) ")
+
     allocate(res(size(mat, dim_)))
 
     if (dim_ == 1) then

src/stdlib_experimental_stat_mean.fypp.f90

@@ -5,6 +5,7 @@
 #:set TYPES = ["real({})".format(k) for k in REALKINDS]
 #:set ikt = list(zip(range(len(KINDS)), KINDS, TYPES))
 
+use stdlib_experimental_error, only: error_stop
 use stdlib_experimental_optval, only: optval
 implicit none
 
@@ -31,6 +32,8 @@ module function mean_2_${k1}$_${k1}$(mat, dim) result(res)
 
     dim_ = optval(dim, 1)
 
+    if (dim_ < 0 .or. dim_ > 2 ) call error_stop("ERROR (mean): invalid argument (dim) ")
+
     allocate(res(size(mat, dim_)))
 
     if (dim_ == 1) then