Юніт-тестування

Принципи Software Engineering у Python

Adam Spannbauer

Machine Learning Engineer at Eastman

Навіщо тестувати?

Перевірити, що код працює як задумано
Гарантувати, що зміни в одній функції не ламають іншу
Захиститися від змін у залежностях

Тестування в Python

doctest
pytest

логотип pytest

Використання doctest

def square(x):
    """Square the number x

    :param x: number to square
    :return: x squared

    >>> square(3)
    9
    """
    return x ** 3



import doctest
doctest.testmod()

Failed example:
    square(3)
Expected:
    9
Got:
    27

Структура pytest

Структура тесту

Структура pytest

Структура тесту

Написання юніт-тестів

робота у workdir/tests/test_document.py

from text_analyzer import Document


# Test tokens attribute on Document object
def test_document_tokens():
    doc = Document('a e i o u')

    assert doc.tokens == ['a', 'e', 'i', 'o', 'u']


# Test edge case of blank document
def test_document_empty():
    doc = Document('')

    assert doc.tokens == []
    assert doc.word_counts == Counter()

Написання юніт-тестів

# Create 2 identical Document objects
doc_a = Document('a e i o u')
doc_b = Document('a e i o u')

# Check if objects are ==
print(doc_a == doc_b)

# Check if attributes are ==
print(doc_a.tokens == doc_b.tokens)
print(doc_a.word_counts == doc_b.word_counts)

False

True
True

Запуск pytest

робота з terminal

datacamp@server:~/work_dir $ pytest

collected 2 items

tests/test_document.py ..                     [100%]

========== 2 passed in 0.61 seconds ==========

Запуск pytest

робота з terminal

datacamp@server:~/work_dir $ pytest tests/test_document.py

collected 2 items

tests/test_document.py ..                     [100%]

========== 2 passed in 0.61 seconds ==========

Падіння тестів

робота з terminal

datacamp@server:~/work_dir $ pytest

collected 2 items

tests/test_document.py F.

============== FAILURES ==============
________ test_document_tokens ________

def test_document_tokens(): doc = Document('a e i o u')

assert doc.tokens == ['a', 'e', 'i', 'o']
E AssertionError: assert ['a', 'e', 'i', 'o', 'u'] == ['a', 'e', 'i', 'o']
E Left contains more items, first extra item: 'u'
E Use -v to get the full diff

tests/test_document.py:7: AssertionError
====== 1 failed in 0.57 seconds ======

Давайте потренуємось

Принципи Software Engineering у Python