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 contents = open(sys.argv[1]).read()
  42 modules = re.findall(mod_regex, contents)
  43 modules = [m.lower()+'.mod' for m in modules]
  44 for t in sys.argv[2:] + modules:
  45     open(t, 'wb').write('myfortran.py wrote %s\n' % os.path.split(t)[1])
  46 sys.exit(0)
  47 """)
  48
  49 test.write('SConstruct', """
  50 env = Environment(FORTRANCOM = r'%(_python_)s myfortran.py $SOURCE $TARGET')
  51 env.Object(target = 'test1.obj', source = 'test1.f')
  52 """ % locals())
  53
  54 test.write('test1.f', """\
  55       PROGRAM TEST
  56       USE MOD_FOO
  57       USE MOD_BAR
  58       PRINT *,'TEST.f'
  59       CALL P
  60       STOP
  61       END
  62       MODULE MOD_FOO
  63          IMPLICIT NONE
  64          CONTAINS
  65          SUBROUTINE P
  66             PRINT *,'mod_foo'
  67          END SUBROUTINE P
  68       END MODULE MOD_FOO
  69       MODULE PROCEDURE MOD_BAR
  70          IMPLICIT NONE
  71          CONTAINS
  72          SUBROUTINE P
  73             PRINT *,'mod_bar'
  74          END SUBROUTINE P
  75       END MODULE MOD_BAR
  76 """)
  77
  78 test.run(arguments = '.', stderr = None)
  79
  80 test.must_match('test1.obj', "myfortran.py wrote test1.obj\n")
  81 test.must_match('mod_foo.mod', "myfortran.py wrote mod_foo.mod\n")
  82 test.must_not_exist('mod_bar.mod')
  83
  84 test.up_to_date(arguments = '.')
  85
  86
  87
  88 test.pass_test()
  89
  90 # Local Variables:
  91 # tab-width:4
  92 # indent-tabs-mode:nil
  93 # End:
  94 # vim: set expandtab tabstop=4 shiftwidth=4: