Python_読み込み、grep、書き込み - Codeまとめby長谷川

【内容】
入力ファイルをgrepして出力ファイルに出力する。

【コード】
import re

input_fileobj = open("in.txt", "r")
output_fileobj = open("tmp.txt", "w", buffering=1)

wspace = r"[ \t]"
wspace_regexp = re.compile(wspace)
my_regexp = re.compile(r"(HWY)")


for line in input_fileobj.readlines():
    line = wspace_regexp.sub(",", line)
    if my_regexp.search(line):
        print(line, end="")
        hoge = output_fileobj.write(line)

output_fileobj.close()
input_fileobj.close()

【テストコード】
import unittest

from parse_file import parse_file

class TestParse_file(unittest.TestCase):
    def testInput1(self):
        #引数なしで例外
        self.assertEqual(parse_file(), None)
    def testInput2(self):
        #入力ファイル空白で例外
        self.assertEqual(parse_file(""),None)
    def testInput3(self):
        #入力ファイルなしで例外
        #ファイルないか確認する処理
        self.assertEqual(parse_file("fjoiajfoa"),None)
        
if __name__=='__main__':
    try:
        unittest.main()
    except SystemExit as inst:
        if inst.args[0] is True: # raised by sys.exit(True) when tests failed
            raise