test/SHF77.py

   1 #!/usr/bin/env python
   2 #
   3 # Copyright (c) 2001, 2002, 2003 Steven Knight
   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 os
  28 import string
  29 import sys
  30 import TestSCons
  31
  32 python = TestSCons.python
  33
  34 if sys.platform == 'win32':
  35     _obj = '.obj'
  36 else:
  37     _obj = '.os'
  38
  39 test = TestSCons.TestSCons()
  40
  41
  42
  43 test.write('myg77.py', r"""
  44 import getopt
  45 import sys
  46 opts, args = getopt.getopt(sys.argv[1:], 'cf:o:')
  47 for opt, arg in opts:
  48     if opt == '-o': out = arg
  49 infile = open(args[0], 'rb')
  50 outfile = open(out, 'wb')
  51 for l in infile.readlines():
  52     if l[:4] != '#g77':
  53         outfile.write(l)
  54 sys.exit(0)
  55 """)
  56
  57
  58
  59 test.write('SConstruct', """
  60 env = Environment(SHF77 = r'%s myg77.py')
  61 env.SharedObject(target = 'test1', source = 'test1.f')
  62 env.SharedObject(target = 'test2', source = 'test2.for')
  63 env.SharedObject(target = 'test3', source = 'test3.FOR')
  64 env.SharedObject(target = 'test4', source = 'test4.F')
  65 env.SharedObject(target = 'test5', source = 'test5.fpp')
  66 env.SharedObject(target = 'test6', source = 'test6.FPP')
  67 """ % python)
  68
  69 test.write('test1.f', r"""This is a .f file.
  70 #g77
  71 """)
  72
  73 test.write('test2.for', r"""This is a .for file.
  74 #g77
  75 """)
  76
  77 test.write('test3.FOR', r"""This is a .FOR file.
  78 #g77
  79 """)
  80
  81 test.write('test4.F', r"""This is a .F file.
  82 #g77
  83 """)
  84
  85 test.write('test5.fpp', r"""This is a .fpp file.
  86 #g77
  87 """)
  88
  89 test.write('test6.FPP', r"""This is a .FPP file.
  90 #g77
  91 """)
  92
  93 test.run(arguments = '.', stderr = None)
  94
  95 test.fail_test(test.read('test1' + _obj) != "This is a .f file.\n")
  96
  97 test.fail_test(test.read('test2' + _obj) != "This is a .for file.\n")
  98
  99 test.fail_test(test.read('test3' + _obj) != "This is a .FOR file.\n")
 100
 101 test.fail_test(test.read('test4' + _obj) != "This is a .F file.\n")
 102
 103 test.fail_test(test.read('test5' + _obj) != "This is a .fpp file.\n")
 104
 105 test.fail_test(test.read('test6' + _obj) != "This is a .FPP file.\n")
 106
 107
 108
 109 g77 = test.where_is('g77')
 110
 111 if g77:
 112
 113     test.write("wrapper.py",
 114 """import os
 115 import string
 116 import sys
 117 open('%s', 'wb').write("wrapper.py\\n")
 118 os.system(string.join(sys.argv[1:], " "))
 119 """ % string.replace(test.workpath('wrapper.out'), '\\', '\\\\'))
 120
 121     test.write('SConstruct', """
 122 foo = Environment(LIBS = 'g2c')
 123 shf77 = foo.Dictionary('SHF77')
 124 bar = foo.Copy(SHF77 = r'%s wrapper.py ' + shf77)
 125 foo.SharedObject(target = 'foo/foo', source = 'foo.f')
 126 bar.SharedObject(target = 'bar/bar', source = 'bar.f')
 127 """ % python)
 128
 129     test.write('foo.f', r"""
 130       PROGRAM FOO
 131       PRINT *,'foo.f'
 132       STOP
 133       END
 134 """)
 135
 136     test.write('bar.f', r"""
 137       PROGRAM BAR
 138       PRINT *,'bar.f'
 139       STOP
 140       END
 141 """)
 142
 143
 144     test.run(arguments = 'foo', stderr = None)
 145
 146     test.fail_test(os.path.exists(test.workpath('wrapper.out')))
 147
 148     test.run(arguments = 'bar')
 149
 150     test.fail_test(test.read('wrapper.out') != "wrapper.py\n")
 151
 152 test.pass_test()