test/Fortran/USE-MODULE.py

   1 #!/usr/bin/env python
   2 #
   3 # __COPYRIGHT__
   4 #
   5 # Permission is hereby granted, free of charge, to any person obtaining
   6 # a copy of this software and associated documentation files (the
   7 # "Software"), to deal in the Software without restriction, including
   8 # without limitation the rights to use, copy, modify, merge, publish,
   9 # distribute, sublicense, and/or sell copies of the Software, and to
  10 # permit persons to whom the Software is furnished to do so, subject to
  11 # the following conditions:
  12 #
  13 # The above copyright notice and this permission notice shall be included
  14 # in all copies or substantial portions of the Software.
  15 #
  16 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY
  17 # KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
  18 # WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  19 # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  20 # LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  21 # OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  22 # WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  23 #
  24
  25 __revision__ = "__FILE__ __REVISION__ __DATE__ __DEVELOPER__"
  26
  27 import TestSCons
  28
  29 _python_ = TestSCons._python_
  30 _exe   = TestSCons._exe
  31
  32 test = TestSCons.TestSCons()
  33
  34
  35
  36 test.write('myfortran.py', r"""
  37 import os.path
  38 import re
  39 import sys
  40 mod_regex = "(?im)^\\s*MODULE\\s+(?!PROCEDURE)(\\w+)"
  41 with open(sys.argv[1]) as f:
  42     contents = f.read()
  43 modules = re.findall(mod_regex, contents)
  44 modules = [m.lower()+'.mod' for m in modules]
  45 for t in sys.argv[2:] + modules:
  46     with open(t, 'wb') as f:
  47         f.write(('myfortran.py wrote %s\n' % os.path.split(t)[1]).encode())
  48 sys.exit(0)
  49 """)
  50
  51 test.write('SConstruct', """
  52 env = Environment(FORTRANCOM = r'%(_python_)s myfortran.py $SOURCE $TARGET')
  53 env.Object(target = 'test1.obj', source = 'test1.f')
  54 """ % locals())
  55
  56 test.write('test1.f', """\
  57       PROGRAM TEST
  58       USE MOD_FOO
  59       USE MOD_BAR
  60       PRINT *,'TEST.f'
  61       CALL P
  62       STOP
  63       END
  64       MODULE MOD_FOO
  65          IMPLICIT NONE
  66          CONTAINS
  67          SUBROUTINE P
  68             PRINT *,'mod_foo'
  69          END SUBROUTINE P
  70       END MODULE MOD_FOO
  71       MODULE PROCEDURE MOD_BAR
  72          IMPLICIT NONE
  73          CONTAINS
  74          SUBROUTINE P
  75             PRINT *,'mod_bar'
  76          END SUBROUTINE P
  77       END MODULE MOD_BAR
  78 """)
  79
  80 test.run(arguments = '.', stderr = None)
  81
  82 test.must_match('test1.obj', "myfortran.py wrote test1.obj\n")
  83 test.must_match('mod_foo.mod', "myfortran.py wrote mod_foo.mod\n")
  84 test.must_not_exist('mod_bar.mod')
  85
  86 test.up_to_date(arguments = '.')
  87
  88
  89
  90 test.pass_test()
  91
  92 # Local Variables:
  93 # tab-width:4
  94 # indent-tabs-mode:nil
  95 # End:
  96 # vim: set expandtab tabstop=4 shiftwidth=4: