Python TDD

unittest

• Python에 포함된 다양한 테스트를 자동화할 수 있는 기능이 포함되어 있는 표준 라이브러리
• unittest에 포함된 주요 개념
  • TestCase : unittest 프레임 워크의 테스트 조직의 기본 단위
  • Fixture :
    • 테스트함수의 전 또는 후에 실행
    • 테스트가 실행되기 전에 테스트 환경이 예상 된 상태에 있는지 확인하는 데 사용
    • 테스트 전에 데이터베이스 테이블을 만들거나 테스트 후에 사용한 리소스를 정리하는데 사용
  • assertion :
    • unittest가 테스트가 통과하는지 또는 실패 하는지를 결정.
    • bool test, 객체의 적합성, 적절한 예외 발생 등 다양한 점검을 할 수 있음
    • assertion이 실패하면 테스트 함수가 실패합니다.

TDD (Test Driven Development) 또는 Test Case를 작성하기 위한 파이썬 내장 모듈입니다.

unittest.TestCase를 상속받아 Custom Test Class를 정의합니다.

test_로 시작하는 함수를 정의하면 unittest가 실행되면서 해당 함수를 테스트합니다.

import unittest

def add_two_numbers(a, b):
    return a + b

class TestAdd(unittest.TestCase):
    def test_numbers_3_4(self):
        # 3+4 must be 7
        self.assertEqual(add_two_numbers(3, 4), 7)

    def test_numbers_0_0(self):
        # 0+0 must be 0
        self.assertEqual(add_two_numbers(0, 0), 0)

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

----------------------------------------------------------------------
Ran 2 tests in 0.001s

OK

Fixture

Fixture란 테스트가 수행되기 이전에 준비과정 혹은 테스트가 종료된 이후에 클린업하는 과정을 의미합니다. 예를 들어, 테스트를 위한 임시 데이터 베이스나 폴더를 생성하고 테스트가 끝난후에 이들을 정리하는 등의 일입니다.

• setUp()
  • 각 테스트 메소드를 호출하기 이전에 호출되는 메소드입니다. 테스트 케이스가 실행될 때 마다 사용됩니다.
• tearDown()
  • 각 테스트가 끝난 이후에 호출되는 메소드입니다. 테스트 과정에서 exception이 발생해도 실행됩니다. setUp 메소드가 성공했을 경우에만 호출됩니다.
• setUpClass()
  • 해당 테스트 클래스가 시작되기 이전 단 한번 호출되는 메소드입니다. setUp메소드는 각 테스트 메소드 이전에 항상 호출 됩니다. 메소드에 @classmethod라는 데코레이터를 달아줘야하고 메소드 인자로 cls를 넘겨줘야합니다.
• tearDownClass()
  • 해당 테스트 클래스가 종료된 이후 단 한번 호출되는 메소드입니다. tearDown메소드는 각 테스트 메소드가 끝난 이후 항상 호출됩니다. 메소드에 @classmethod라는 데코레이터를 달아줘야하고 메소드 인자로 cls를 넘겨줘야합니다.

import unittest

def remove_blank(text):
    return text.replace(' ', '')

def remove_new_line(text):
    return text.replace('\\n', '')

def add_semicolon(text):
    return text + ';'

class TestFixture(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        print('setUpClass')

    @classmethod
    def tearDownClass(cls):
        print('tearDownClass')

    def setUp(self):
        print('setUp')

    def tearDown(self):
        print('tearDown')

    def test_remove_blank(self):
        text = "Hi, I'm SJ Quant"
        removed = remove_blank(text)
        self.assertNotIn(' ', removed)

    def test_remove_new_line(self):
        text = "Hi, I'm SJ Quant\\nNice to meet you."
        removed = remove_new_line(text)
        self.assertNotIn('\\n', removed)

    def test_add_semicolon(self):
        text = "console.log('Hello, World')"
        added = add_semicolon(text)
        self.assertEqual(added, "console.log('Hello, World');")

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

setUpClass
setUp
tearDown
setUp
tearDown
setUp
tearDown
tearDownClass

----------------------------------------------------------------------Ran 3 tests in 0.005s

OK

setUpClass와 tearDownClass메소드는 각각 시작과 끝에 한 번씩 실행되었고, setUp과 tearDown메소드는 테스트 메소드 개수와 동일하게 3번씩 실행 된 것을 확인 할 수 있습니다.