test/Fortran/FORTRAN.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 string
  28
  29 import TestSCons
  30
  31 from common import write_fake_link
  32
  33 _python_ = TestSCons._python_
  34 _exe   = TestSCons._exe
  35
  36 test = TestSCons.TestSCons()
  37
  38 write_fake_link(test)
  39
  40 test.write('myg77.py', r"""
  41 import getopt
  42 import sys
  43 opts, args = getopt.getopt(sys.argv[1:], 'co:')
  44 for opt, arg in opts:
  45     if opt == '-o': out = arg
  46 infile = open(args[0], 'rb')
  47 outfile = open(out, 'wb')
  48 for l in infile.readlines():
  49     if l[:4] != '#g77':
  50         outfile.write(l)
  51 sys.exit(0)
  52 """)
  53
  54 test.write('SConstruct', """
  55 env = Environment(LINK = r'%(_python_)s mylink.py',
  56                   LINKFLAGS = [],
  57                   FORTRAN = r'%(_python_)s myg77.py')
  58 env.Program(target = 'test01', source = 'test01.f')
  59 env.Program(target = 'test02', source = 'test02.F')
  60 env.Program(target = 'test03', source = 'test03.for')
  61 env.Program(target = 'test04', source = 'test04.FOR')
  62 env.Program(target = 'test05', source = 'test05.ftn')
  63 env.Program(target = 'test06', source = 'test06.FTN')
  64 env.Program(target = 'test07', source = 'test07.fpp')
  65 env.Program(target = 'test08', source = 'test08.FPP')
  66 """ % locals())
  67
  68 test.write('test01.f',   "This is a .f file.\n#link\n#g77\n")
  69 test.write('test02.F',   "This is a .F file.\n#link\n#g77\n")
  70 test.write('test03.for', "This is a .for file.\n#link\n#g77\n")
  71 test.write('test04.FOR', "This is a .FOR file.\n#link\n#g77\n")
  72 test.write('test05.ftn', "This is a .ftn file.\n#link\n#g77\n")
  73 test.write('test06.FTN', "This is a .FTN file.\n#link\n#g77\n")
  74 test.write('test07.fpp', "This is a .fpp file.\n#link\n#g77\n")
  75 test.write('test08.FPP', "This is a .FPP file.\n#link\n#g77\n")
  76
  77 test.run(arguments = '.', stderr = None)
  78
  79 test.must_match('test01' + _exe, "This is a .f file.\n")
  80 test.must_match('test02' + _exe, "This is a .F file.\n")
  81 test.must_match('test03' + _exe, "This is a .for file.\n")
  82 test.must_match('test04' + _exe, "This is a .FOR file.\n")
  83 test.must_match('test05' + _exe, "This is a .ftn file.\n")
  84 test.must_match('test06' + _exe, "This is a .FTN file.\n")
  85 test.must_match('test07' + _exe, "This is a .fpp file.\n")
  86 test.must_match('test08' + _exe, "This is a .FPP file.\n")
  87
  88
  89
  90 fc = 'f77'
  91 f77 = test.detect_tool(fc)
  92 FTN_LIB = TestSCons.fortran_lib
  93
  94 if f77:
  95
  96     test.write("wrapper.py",
  97 """import os
  98 import string
  99 import sys
 100 open('%s', 'wb').write("wrapper.py\\n")
 101 os.system(string.join(sys.argv[1:], " "))
 102 """ % string.replace(test.workpath('wrapper.out'), '\\', '\\\\'))
 103
 104     test.write('SConstruct', """
 105 foo = Environment(FORTRAN = '%(fc)s')
 106 f77 = foo.Dictionary('FORTRAN')
 107 bar = foo.Clone(FORTRAN = r'%(_python_)s wrapper.py ' + f77)
 108 foo.Program(target = 'foo', source = 'foo.f')
 109 bar.Program(target = 'bar', source = 'bar.f')
 110 """ % locals())
 111
 112     test.write('foo.f', r"""
 113       PROGRAM FOO
 114       PRINT *,'foo.f'
 115       STOP
 116       END
 117 """)
 118
 119     test.write('bar.f', r"""
 120       PROGRAM BAR
 121       PRINT *,'bar.f'
 122       STOP
 123       END
 124 """)
 125
 126
 127     test.run(arguments = 'foo' + _exe, stderr = None)
 128
 129     test.run(program = test.workpath('foo'), stdout =  " foo.f\n")
 130
 131     test.must_not_exist('wrapper.out')
 132
 133     import sys
 134     if sys.platform[:5] == 'sunos':
 135         test.run(arguments = 'bar' + _exe, stderr = None)
 136     else:
 137         test.run(arguments = 'bar' + _exe)
 138
 139     test.run(program = test.workpath('bar'), stdout =  " bar.f\n")
 140
 141     test.must_match('wrapper.out', "wrapper.py\n")
 142
 143 test.pass_test()
 144
 145 # Local Variables:
 146 # tab-width:4
 147 # indent-tabs-mode:nil
 148 # End:
 149 # vim: set expandtab tabstop=4 shiftwidth=4: