# file_handler.py
#-*- coding: utf-8 -*-


# 테스트할 함수 : 파일 경로를 입력하면 해당 파일의 라인수를 반환
def file_line_counter(file_path):
    with open(file_path,"r") as f:
        lines = f.readlines()

    return sum(1 for line in lines)

#-*- coding: utf-8 -*-

import unittest
import os
import file_handler


class FileHandlerTest(unittest.TestCase):
    testfile = "./data/test"

    # setUp : 유닛테스트 시작하기 전에 실행되는 함수 (테스트 파일 생성)
    def setUp(self):
        with open(self.testfile, "w") as f:
            f.write('1\n2\n3\n4\n5\n6')

    # tearDown : 유닛테스트 종료 후 실행되는 함수 (테스트 파일 삭제)
    def tearDown(self):
        try:
            os.remove(self.testfile)
        except:
            pass

    # 테스트 함수 (arrsertEqual 메소드를 통해 기대하는 결과가 나오는지 테스트)
    def test_file_line_counter(self):
        self.assertEqual(file_handler.file_line_counter(self.testfile), 6)

if __name__ == "__main__":
    unittest.main()

.
----------------------------------------------------------------------
Ran 1 test in 0.000s

OK

======================================================================
FAIL: test_file_line_counter (__main__.FileHandlerTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "D:/git_store/ODOC/python/unittest_.py", line 28, in test_file_line_counter
    self.assertEqual(file_handler.file_line_counter(self.testfile), 5)
AssertionError: 6 != 5

----------------------------------------------------------------------
Ran 1 test in 0.000s

FAILED (failures=1)