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