Turn floating point tests to use tolerance

Author: aradi

src/tests/stats/test_mean.f90

@@ -5,6 +5,9 @@ program test_mean
 use stdlib_experimental_stats, only: mean
 implicit none
 
+real(sp), parameter :: sptol = 1.2e-06_sp
+real(dp), parameter :: dptol = 2.2e-15_dp
+
 real(sp) :: s1(3) = [1.0_sp, 2.0_sp, 3.0_sp]
 
 real(sp), allocatable :: s(:, :)
@@ -17,37 +20,37 @@ program test_mean
 !sp
 call loadtxt("array3.dat", s)
 
-call assert( mean(s) - sum(s)/real(size(s), sp) == 0.0_sp)
-call assert( sum( abs( mean(s,1) - sum(s,1)/real(size(s,1), sp) )) == 0.0_sp)
-call assert( sum( abs( mean(s,2) - sum(s,2)/real(size(s,2), sp) )) == 0.0_sp)
+call assert( abs(mean(s) - sum(s)/real(size(s), sp)) < sptol)
+call assert( sum( abs( mean(s,1) - sum(s,1)/real(size(s,1), sp) )) < sptol)
+call assert( sum( abs( mean(s,2) - sum(s,2)/real(size(s,2), sp) )) < sptol)
 
 ! check reduction of rank one array to scalar
-call assert(mean(s1) - sum(s1) / real(size(s1), sp) == 0.0_dp)
-call assert(mean(s1, dim=1) - sum(s1, dim=1) / real(size(s1, dim=1), sp) == 0.0_dp)
+call assert(abs(mean(s1) - sum(s1) / real(size(s1), sp)) < sptol)
+call assert(abs(mean(s1, dim=1) - sum(s1, dim=1) / real(size(s1, dim=1), sp)) < sptol)
 
 
 !dp
 call loadtxt("array3.dat", d)
 
-call assert( mean(d) - sum(d)/real(size(d), dp) == 0.0_dp)
-call assert( sum( abs( mean(d,1) - sum(d,1)/real(size(d,1), dp) )) == 0.0_dp)
-call assert( sum( abs( mean(d,2) - sum(d,2)/real(size(d,2), dp) )) == 0.0_dp)
+call assert( abs(mean(d) - sum(d)/real(size(d), dp)) < dptol)
+call assert( sum( abs( mean(d,1) - sum(d,1)/real(size(d,1), dp) )) < dptol)
+call assert( sum( abs( mean(d,2) - sum(d,2)/real(size(d,2), dp) )) < dptol)
 
 
 !int32
 call loadtxt("array3.dat", d)
 
-call assert( mean(int(d, int32)) - sum(real(int(d, int32),dp))/real(size(d), dp) == 0.0_dp)
-call assert( sum(abs( mean(int(d, int32),1) - sum(real(int(d, int32),dp),1)/real(size(d,1), dp) )) == 0.0_dp)
-call assert( sum(abs( mean(int(d, int32),2) - sum(real(int(d, int32),dp),2)/real(size(d,2), dp) )) == 0.0_dp)
+call assert( abs(mean(int(d, int32)) - sum(real(int(d, int32),dp))/real(size(d), dp)) < dptol)
+call assert( sum(abs( mean(int(d, int32),1) - sum(real(int(d, int32),dp),1)/real(size(d,1), dp) )) < dptol)
+call assert( sum(abs( mean(int(d, int32),2) - sum(real(int(d, int32),dp),2)/real(size(d,2), dp) )) < dptol)
 
 
 !int64
 call loadtxt("array3.dat", d)
 
-call assert( mean(int(d, int64)) - sum(real(int(d, int64),dp))/real(size(d), dp) == 0.0_dp)
-call assert( sum(abs( mean(int(d, int64),1) - sum(real(int(d, int64),dp),1)/real(size(d,1), dp) )) == 0.0_dp)
-call assert( sum(abs( mean(int(d, int64),2) - sum(real(int(d, int64),dp),2)/real(size(d,2), dp) )) == 0.0_dp)
+call assert( abs(mean(int(d, int64)) - sum(real(int(d, int64),dp))/real(size(d), dp)) < dptol)
+call assert( sum(abs( mean(int(d, int64),1) - sum(real(int(d, int64),dp),1)/real(size(d,1), dp) )) < dptol)
+call assert( sum(abs( mean(int(d, int64),2) - sum(real(int(d, int64),dp),2)/real(size(d,2), dp) )) < dptol)
 
 
 !dp rank 3
@@ -56,10 +59,10 @@ program test_mean
 d3(:,:,2)=d*1.5_dp;
 d3(:,:,3)=d*4._dp;
 
-call assert( mean(d3) - sum(d3)/real(size(d3), dp) == 0.0_dp)
-call assert( sum( abs( mean(d3,1) - sum(d3,1)/real(size(d3,1), dp) )) == 0.0_dp)
-call assert( sum( abs( mean(d3,2) - sum(d3,2)/real(size(d3,2), dp) )) == 0.0_dp)
-call assert( sum( abs( mean(d3,3) - sum(d3,3)/real(size(d3,3), dp) )) == 0.0_dp)
+call assert( abs(mean(d3) - sum(d3)/real(size(d3), dp)) < dptol)
+call assert( sum( abs( mean(d3,1) - sum(d3,1)/real(size(d3,1), dp) )) < dptol)
+call assert( sum( abs( mean(d3,2) - sum(d3,2)/real(size(d3,2), dp) )) < dptol)
+call assert( sum( abs( mean(d3,3) - sum(d3,3)/real(size(d3,3), dp) )) < dptol)
 
 
 !dp rank 4
@@ -70,12 +73,10 @@ program test_mean
 d4(:,:,3,1)=d*4._dp;
 d4(:,:,3,9)=d*4._dp;
 
-call assert( mean(d4) - sum(d4)/real(size(d4), dp) == 0.0_dp)
-call assert( sum( abs( mean(d4,1) - sum(d4,1)/real(size(d4,1), dp) )) == 0.0_dp)
-call assert( sum( abs( mean(d4,2) - sum(d4,2)/real(size(d4,2), dp) )) == 0.0_dp)
-call assert( sum( abs( mean(d4,3) - sum(d4,3)/real(size(d4,3), dp) )) == 0.0_dp)
-call assert( sum( abs( mean(d4,4) - sum(d4,4)/real(size(d4,4), dp) )) == 0.0_dp)
-
-contains
+call assert( abs(mean(d4) - sum(d4)/real(size(d4), dp)) < dptol)
+call assert( sum( abs( mean(d4,1) - sum(d4,1)/real(size(d4,1), dp) )) < dptol)
+call assert( sum( abs( mean(d4,2) - sum(d4,2)/real(size(d4,2), dp) )) < dptol)
+call assert( sum( abs( mean(d4,3) - sum(d4,3)/real(size(d4,3), dp) )) < dptol)
+call assert( sum( abs( mean(d4,4) - sum(d4,4)/real(size(d4,4), dp) )) < dptol)
 
 end program