From 7534cedb89baada863bf48f51b6a1635dfa6cee8 Mon Sep 17 00:00:00 2001
From: sgeard <sgeard@users.noreply.github.com>
Date: Sat, 10 Jun 2023 12:03:55 +0100
Subject: [PATCH] Improved testing of amap

---
 test/test_amap.f90 | 26 ++++++++++++++++++++++----
 1 file changed, 22 insertions(+), 4 deletions(-)

diff --git a/test/test_amap.f90 b/test/test_amap.f90
index c570c7f..7fb0361 100644
--- a/test/test_amap.f90
+++ b/test/test_amap.f90
@@ -1,18 +1,36 @@
 program test_amap
     use amap
+    use iso_fortran_env, only: output_unit
     implicit none
     
     type(amap_t) :: my_amap
     type(value_t) :: x
     
+    write(output_unit,fmt='(a)',advance='no') 'checking set ... '
     call my_amap%set('one',1.0d0)
     call my_amap%set('two',2.d0)
     call my_amap%set('three',3.0d0)
+    call my_amap%set('four',7.0d0)
     call my_amap%set('four',4.0d0)
     call my_amap%set('five',5.0d0)
-    call my_amap%print
+    if (my_amap%size() == 5) then
+        write(output_unit,'(a)') 'passed'
+    else
+        write(output_unit,'(a)') 'FAILED'
+    end if
     
-    x = my_amap%get('four')
-    write(6,'(f0.6)') my_amap%get_value('four')
-    print *,my_amap%contains('one'),my_amap%contains('ten')
+    write(output_unit,fmt='(a)',advance='no') 'checking existence ... '
+    if (my_amap%contains('one') .and. .not. my_amap%contains('ten')) then
+        write(output_unit,'(a)') 'passed'
+    else
+        write(output_unit,'(a)') 'FAILED'
+    end if
+    
+    write(output_unit,fmt='(a)',advance='no') 'checking data ... '
+    if (my_amap%get('four') == 4.0d0) then
+        write(output_unit,'(a)') 'passed'
+    else
+        write(output_unit,'(a)') 'FAILED'
+    end if
+
 end program test_amap