Accessing Python dictionaries within fypp

Is it possible to assign and retrieve Python dictionary key-value pairs from within fypp without writing a dedicated Python class? If yes, what is the correct syntax to do so?

[aradi]

I am happy you managed. Just a comment: Variables in the global scope (chkval, tslb1, etc.) do not need a #:global assignment, they are global by default…

Otherwise, you may eventually be interested in the FyTest unit testing framework, which offers unit testing and uses the aforementioned dictionary manipulation quite heavily.

[wyphan]

Never mind, I figured it out! Here’s my current mod_testing_macros.fypp:

#! -*- mode: f90; -*-
#:mute
!===============================================================================
! Testing infrastructure fypp macros
! Last edited: Nov 10, 2021 (WYP)
!===============================================================================
#:endmute

#! Initialize testing infrastructure

#! Number of tests
#:global ntst
#:set ntst = 0

#! Dictionaries to hold golden values and test labels
#:global chkval
#:set chkval = {}
#:global tstlbl
#:set tstlbl = {}

#! Macro to register golden value
#:def save_gold( lbl, val )
  $:tstlbl.update([(ntst, lbl)])
  $:chkval.update([(ntst, val)])
#:enddef save_gold

#! Macro to check scalar test result exactly against golden value
#:def check_exact( val )
  #:set lbl  = tstlbl[ntst]
  #:set gold = chkval[ntst]
  BLOCK
    USE ISO_FORTRAN_ENV, ONLY: o => output_unit, e => error_unit
    WRITE(o,'(A)') ${lbl}$
    IF( ${val}$ == ${gold}$ ) THEN
      WRITE(e,*) 'OK: result = ', ${val}$
    ELSE
      nerr = nerr + 1
      WRITE(e,*) 'Error[${_FILE_}$]: result = ', ${val}$, ', should be ${gold}$'
    END IF
  END BLOCK
  #:set ntst = ntst + 1
#:enddef check_exact

#:def test_summary()
  BLOCK
    USE ISO_FORTRAN_ENV, ONLY: o => output_unit
    WRITE(o,'("Summary for ${_FILE_}$:")')
    IF( nerr > 0 ) THEN
      WRITE(o,'("  ",I0," out of ${ntst}$ tests failed")') nerr
    ELSE
      WRITE(o,'("  All ${ntst}$ tests succeeded")')
    END IF
  END BLOCK
#:enddef test_summary