def test_filename_reader():
'''
Tests that a Fortran source file can be read given its filename.
'''
handle, filename = tempfile.mkstemp(suffix='.f90', text=True)
os.close(handle)
try:
with io.open(filename, mode='w', encoding='UTF-8') as source_file:
source_file.write(FULL_FREE_SOURCE)
unit_under_test = FortranFileReader(filename)
expected = fparser.common.sourceinfo.FortranFormat(True, False)
assert unit_under_test.format == expected
for expected in FULL_FREE_EXPECTED:
found = unit_under_test.get_single_line(ignore_empty=True)
assert found == expected
except Exception:
os.unlink(filename)
raise
def test_file_reader():
'''
Tests that a Fortran source file can be read given a file object of it.
'''
handle, filename = tempfile.mkstemp(suffix='.f90', text=True)
os.close(handle)
try:
with open(filename, mode='w') as source_file:
print(FULL_FREE_SOURCE, file=source_file)
with open(filename, mode='r') as source_file:
unit_under_test = FortranFileReader(source_file)
expected = fparser.common.sourceinfo.FortranFormat(True, False)
assert unit_under_test.format == expected
for expected in FULL_FREE_EXPECTED:
assert unit_under_test.get_single_line(ignore_empty=True) \
== expected
except Exception:
os.unlink(filename)
raise