Basic tests for maxpool2d

Author: milancurcic

CMakeLists.txt

@@ -100,7 +100,7 @@ string(REGEX REPLACE "^ | $" "" LIBS "${LIBS}")
 
 # tests
 enable_testing()
-foreach(execid input1d_layer dense_layer conv2d_layer dense_network)
+foreach(execid input1d_layer dense_layer conv2d_layer maxpool2d_layer dense_network)
   add_executable(test_${execid} test/test_${execid}.f90)
   target_link_libraries(test_${execid} neural ${LIBS})
   add_test(test_${execid} bin/test_${execid})

test/test_maxpool2d_layer.f90

@@ -0,0 +1,33 @@
+program test_maxpool2d_layer
+
+  use iso_fortran_env, only: stderr => error_unit
+  use nf, only: maxpool2d, input, layer
+  use nf_input3d_layer, only: input3d_layer
+
+  implicit none
+
+  type(layer) :: maxpool_layer, input_layer
+  integer, parameter :: pool_size = 2, stride = 2
+  real, allocatable :: sample_input(:,:,:), output(:,:,:)
+  logical :: ok = .true.
+
+  maxpool_layer = maxpool2d(pool_size)
+
+  if (.not. maxpool_layer % name == 'maxpool2d') then
+    ok = .false.
+    write(stderr, '(a)') 'maxpool2d layer has its name set correctly.. failed'
+  end if
+
+  if (maxpool_layer % initialized) then
+    ok = .false.
+    write(stderr, '(a)') 'maxpool2d layer should not be marked as initialized yet.. failed'
+  end if
+
+  if (ok) then
+    print '(a)', 'test_maxpool2d_layer: All tests passed.'
+  else
+    write(stderr, '(a)') 'test_maxpool2d_layer: One or more tests failed.'
+    stop 1
+  end if
+
+end program test_maxpool2d_layer