Современный Python: пишем модульные тесты и применяем соглашения о коммитах в Git

Моя цель - предложение широкого ассортимента товаров и услуг на постоянно высоком качестве обслуживания по самым выгодным ценам.

Перевод был сделан для платформы курсов по программированию Хекслет

Эта вторая статья из цикла, в котором рассказывается о лучших практиках современного Python. В этом цикле статей все примеры основаны на реализации простого проекта, который представляет собой функцию Python, которая суммирует данные, присутствующие в pandas DataFrame. Функция выводит количество строк и столбцов и частоту каждого типа данных, присутствующих в pandas DataFrame.

Хорошая практика разработки программного обеспечения всегда приносит много долгосрочных выгод. Например, написание модульных тестов позволяет поддерживать большие кодовые базы и гарантирует, что определенный фрагмент вашего кода ведет себя так, как ожидается. Написание последовательных коммитов в Git также улучшает сотрудничество между участниками проекта. Хорошо составленные сообщения коммитов Git открывают путь к автоматическому версионированию и автоматически генерируемым файлам журнала изменений. Поэтому сейчас в сообществе разработчиков предпринимается множество попыток нормализовать сообщения, написанные в наших Git-коммитах.

В первой части этой серии мы настраивали наш проект, устанавливая различные версии Python с помощью pyenv, устанавливая локальную версию Python с помощью pyenv, инкапсулируя его в виртуальную среду с помощью poetry. Здесь мы покажем более подробно, как проводить модульное тестирование вашего Python-приложения и как обеспечить и проверить сообщения о коммитах в Git. Исходный код, связанный с этой статьей, опубликован на GitHub.

Тестирование кода

Перейдите в корневой каталог вашего проекта и активируйте виртуальное окружение:

cd summarize_dataframe/
poetry shell

Примечание переводчика: Поскольку со времени написания оригинальной статьи требования numpy к версии Python изменились, перед добавлением зависимостей необходимо внести изменения в файл pyproject.toml:

[tool.poetry.dependencies]
-python = "^3.8"
+python = ">=3.8,<3.11"

Добавим несколько зависимостей с помощью poetry:

poetry add -D pynvim numpy pandas

Флаг -D указывает, что зависимость применяется только к среде разработки.

Исходя из ожидаемого результата работы проекта, наша программа состоит из трех шагов:

  1. Получение формы pandas DataFrame

  2. Получение частоты типов pandas dtypes.

  3. Объединение этих двух результатов в единый DataFrame, который будет использован для вывода окончательного результата.

После получения окончательного DataFrame выводится результат, как показано выше. В связи с этим наш код может выглядеть следующим образом:

import pandas as pd
def data_summary(df: pd.DataFrame) -> None:
"""
Function defined to return a DataFrame containing details
about the number of rows and columns and the column dtype
frequency of the passed pandas DataFrame
"""
def _shape(df: pd.DataFrame) -> None:
"""
Function defined to return a dataframe with details about
the number of row and columns
"""
return None
<span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">def</span> <span style="box-sizing: border-box; color: rgb(153, 0, 0); font-weight: bold;">_dtypes_freq</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">df</span><span style="box-sizing: border-box;">:</span> <span style="box-sizing: border-box;">pd</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">DataFrame</span><span style="box-sizing: border-box;">)</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">-&gt;</span> <span style="box-sizing: border-box; color: rgb(153, 153, 153);">None</span><span style="box-sizing: border-box;">:</span>
    <span style="box-sizing: border-box; color: rgb(221, 17, 68);">"""
    Function defined to return a dataframe with details about
    the pandas dtypes frequency
    """</span>
<span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">return</span> <span style="box-sizing: border-box; color: rgb(153, 153, 153);">None</span>
<span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">return</span> <span style="box-sizing: border-box; color: rgb(153, 153, 153);">None</span>

def display_summary(df: pd.DataFrame) -> None:
"""
Function define to print out the result of the data summary
"""
result_df = True
message = '---- Data summary ----'
print(message, result_df, sep='\n')

Теперь начнем писать модульные тесты. Мы будем использовать инструмент unittest, доступный в стандартной библиотеке Python. Возможно вы помните, что в предыдущей статье pytest был определен как зависимость для тестирования. Это не проблема, потому что pytest нативно запускает тесты, написанные с помощью библиотеки unittest.

Юнит-тесты — это методы, которые, как ожидает unittest, будут описаны внутри классов Python. Выберите для своих тестовых классов и методов описывающее имя — оно должно начинаться с test_. Дополнительно unittest использует ряд специальных тестовых методов, унаследованных от класса unittest.TestCase.

На практике тест должен:

  • Охватывать одну функцию

  • Быть автономным

  • Не требовать внешних инструкций

  • Воссоздавать условия достижения результата.

Чтобы воссоздать необходимую рабочую среду, необходимо написать код настройки. Если этот код окажется избыточным, реализуйте метод setUp(), который будет выполняться перед каждым тестом. Это очень удобно для повторного использования и реорганизации кода. В зависимости от сценария использования, возможно придется выполнять регулярные операции после выполнения тестов. Для этого вы можете использовать метод tearDown().

Сначала вы можете посмотреть unit-тест, который мы реализовали для функции data_summary():

import unittest
import pandas as pd
from summarize_dataframe.summarize_df import data_summary
class TestDataSummary(unittest.TestCase):
def setUp(self):
# initialize dataframe to test
        df_data = [[1, 'a'], [2, 'b'], [3, 'c']]
df_cols = ['numbers', 'letters']
self.df = pd.DataFrame(data=df_data, columns=df_cols)
# initialize expected dataframe
        exp_col = ['Values']
exp_idx = ['Number of rows', 'Number of columns', 'int64', 'object']
exp_data = [[3], [2], [1], [1]]
self.exp_df = pd.DataFrame(data=exp_data, columns=exp_col, index=exp_idx)
<span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">def</span> <span style="box-sizing: border-box; color: rgb(153, 0, 0); font-weight: bold;">test_data_summary</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box; color: rgb(153, 153, 153);">self</span><span style="box-sizing: border-box;">):</span>
    <span style="box-sizing: border-box;">expected_df</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> <span style="box-sizing: border-box; color: rgb(153, 153, 153);">self</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">exp_df</span>
    <span style="box-sizing: border-box;">result_df</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> <span style="box-sizing: border-box;">data_summary</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box; color: rgb(153, 153, 153);">self</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">df</span><span style="box-sizing: border-box;">)</span>
    <span style="box-sizing: border-box; color: rgb(153, 153, 153);">self</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">assertTrue</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">expected_df</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">equals</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">result_df</span><span style="box-sizing: border-box;">))</span>

if name == 'main':
unittest.main()

Метод setUp() инициализирует два разных pandas DataFrame. self.exp_df — это результирующий DataFrame, который мы ожидаем получить после вызова функции data_summary(), а self.df — это DataFrame, который используется для тестирования наших функций. Сейчас ожидается, что тесты окажутся неудачными, потому что логика не была реализована. Для тестирования с помощью poetry используйте команду:

poetry run pytest -v

==================================================== test session starts ====================================================
platform darwin -- Python 3.8.7, pytest-5.4.3, py-1.11.0, pluggy-0.13.1 -- /Users/aabur/Documents/GitHub/AABur/modern_python/summarize_dataframe/.venv/bin/python
cachedir: .pytest_cache
rootdir: /Users/aabur/Documents/GitHub/AABur/modern_python/summarize_dataframe
collected 1 item
tests/test_summarize_dataframe.py::TestDataSummary::test_data_summary FAILED                                          [100%]
========================================================= FAILURES ==========================================================
_____________________________________________ TestDataSummary.test_data_summary _____________________________________________
self =
def test_data_summary<span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">(</span>self<span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">)</span>:
    expected_df <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> self.exp_df
    result_df <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> data_summary<span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">(</span>self.df<span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">)</span>

>       self.assertTrue(expected_df.equals(result_df))
E       AssertionError: False is not true
tests/test_summarize_dataframe.py:21: AssertionError
================================================== short test summary info ==================================================
FAILED tests/test_summarize_dataframe.py::TestDataSummary::test_data_summary - AssertionError: False is not true
===================================================== 1 failed in 0.39s =====================================================

Использование флага -v позволяет получить более подробный вывод результатов тестирования. Вы можете видеть, что тесты помечены в соответствии с именами классов и функций, которые вы задали. В нашем случае это ::TestDataSummary::test_data_summary.

Поменяем код для соответствия unit-тестам:

import pandas as pd
def data_summary(df: pd.DataFrame) -> pd.DataFrame:
"""
Function defined to output details about the number
of rows and columns and the column dtype frequency of
the passed pandas DataFrame
"""
def _shape(df: pd.DataFrame) -> pd.DataFrame:
"""
Function defined to return a dataframe with details about
the number of row and columns
"""
row, col = df.shape
return pd.DataFrame(data=[[row], [col]], columns=['Values'], index=['Number of rows', 'Number of columns'])
<span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">def</span> <span style="box-sizing: border-box; color: rgb(153, 0, 0); font-weight: bold;">_dtypes_freq</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">df</span><span style="box-sizing: border-box;">:</span> <span style="box-sizing: border-box;">pd</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">DataFrame</span><span style="box-sizing: border-box;">)</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">-&gt;</span> <span style="box-sizing: border-box;">pd</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">DataFrame</span><span style="box-sizing: border-box;">:</span>
    <span style="box-sizing: border-box; color: rgb(221, 17, 68);">"""
    Function defined to return a dataframe with details about
    the pandas dtypes frequency
    """</span>
    <span style="box-sizing: border-box;">counter</span><span style="box-sizing: border-box;">,</span> <span style="box-sizing: border-box;">types</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> <span style="box-sizing: border-box;">{},</span> <span style="box-sizing: border-box;">df</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">dtypes</span>
    <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">for</span> <span style="box-sizing: border-box;">dtype</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">in</span> <span style="box-sizing: border-box;">types</span><span style="box-sizing: border-box;">:</span>
        <span style="box-sizing: border-box;">tmp</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> <span style="box-sizing: border-box; color: rgb(0, 134, 179);">str</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">dtype</span><span style="box-sizing: border-box;">)</span>
        <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">if</span> <span style="box-sizing: border-box;">tmp</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">in</span> <span style="box-sizing: border-box;">counter</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">keys</span><span style="box-sizing: border-box;">():</span>
            <span style="box-sizing: border-box;">counter</span><span style="box-sizing: border-box;">[</span><span style="box-sizing: border-box;">tmp</span><span style="box-sizing: border-box;">]</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">+=</span> <span style="box-sizing: border-box; color: rgb(0, 153, 153);">1</span>
        <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">else</span><span style="box-sizing: border-box;">:</span>
            <span style="box-sizing: border-box;">counter</span><span style="box-sizing: border-box;">[</span><span style="box-sizing: border-box;">tmp</span><span style="box-sizing: border-box;">]</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> <span style="box-sizing: border-box; color: rgb(0, 153, 153);">1</span>
    <span style="box-sizing: border-box;">values</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> <span style="box-sizing: border-box;">[[</span><span style="box-sizing: border-box;">value</span><span style="box-sizing: border-box;">]</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">for</span> <span style="box-sizing: border-box;">value</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">in</span> <span style="box-sizing: border-box;">counter</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">values</span><span style="box-sizing: border-box;">()]</span>
    <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">return</span> <span style="box-sizing: border-box;">pd</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">DataFrame</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">data</span><span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span><span style="box-sizing: border-box;">values</span><span style="box-sizing: border-box;">,</span> <span style="box-sizing: border-box;">columns</span><span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span><span style="box-sizing: border-box;">[</span><span style="box-sizing: border-box; color: rgb(221, 17, 68);">'Values'</span><span style="box-sizing: border-box;">],</span> <span style="box-sizing: border-box;">index</span><span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span><span style="box-sizing: border-box; color: rgb(0, 134, 179);">list</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">counter</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">keys</span><span style="box-sizing: border-box;">()))</span>
<span style="box-sizing: border-box;">result_df</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> <span style="box-sizing: border-box;">pd</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">concat</span><span style="box-sizing: border-box;">([</span><span style="box-sizing: border-box;">_shape</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">df</span><span style="box-sizing: border-box;">),</span> <span style="box-sizing: border-box;">_dtypes_freq</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">df</span><span style="box-sizing: border-box;">)])</span>
<span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">return</span> <span style="box-sizing: border-box;">result_df</span>

def display_summary(df: pd.DataFrame) -> None:
"""
Function define to print out the result of the data summary
"""
result_df = True
message = '---- Data summary ----'
print(message, result_df, sep='\n')

Снова запустим тесты:

poetry run pytest -v

==================================================== test session starts ====================================================
platform darwin -- Python 3.8.7, pytest-5.4.3, py-1.11.0, pluggy-0.13.1 -- /Users/aabur/Documents/GitHub/AABur/modern_python/summarize_dataframe/.venv/bin/python
cachedir: .pytest_cache
rootdir: /Users/aabur/Documents/GitHub/AABur/modern_python/summarize_dataframe
collected 1 item
tests/test_summarize_dataframe.py::TestDataSummary::test_data_summary PASSED                                          [100%]
===================================================== 1 passed in 0.35s =====================================================

И последнее. В наших тестах мы не проверяли фактический вывод. Наш модуль предназначен для вывода строкового представления сводки DataFrame. Существуют решения для достижения этой цели с помощью unittest, но мы будем использовать pytest для этого теста. Удивительно, не правда ли? Как уже говорилось, pytest очень хорошо работает с unittest, и сейчас мы это проиллюстрируем. Вот код для этого теста:

import unittest
import pytest
import pandas as pd
from summarize_dataframe.summarize_df import data_summary, display_summary
class TestDataSummary(unittest.TestCase):
def setUp(self):
# initialize dataframe to test
        df_data = [[1, 'a'], [2, 'b'], [3, 'c']]
df_cols = ['numbers', 'letters']
self.df = pd.DataFrame(data=df_data, columns=df_cols)
# initialize expected dataframe
        exp_col = ['Values']
exp_idx = ['Number of rows', 'Number of columns', 'int64', 'object']
exp_data = [[3], [2], [1], [1]]
self.exp_df = pd.DataFrame(data=exp_data, columns=exp_col, index=exp_idx)
<span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">@</span><span style="box-sizing: border-box;">pytest</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">fixture</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">autouse</span><span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span><span style="box-sizing: border-box; color: rgb(153, 153, 153);">True</span><span style="box-sizing: border-box;">)</span>
<span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">def</span> <span style="box-sizing: border-box; color: rgb(153, 0, 0); font-weight: bold;">_pass_fixture</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box; color: rgb(153, 153, 153);">self</span><span style="box-sizing: border-box;">,</span> <span style="box-sizing: border-box;">capsys</span><span style="box-sizing: border-box;">):</span>
    <span style="box-sizing: border-box; color: rgb(153, 153, 153);">self</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">capsys</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> <span style="box-sizing: border-box;">capsys</span>

<span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">def</span> <span style="box-sizing: border-box; color: rgb(153, 0, 0); font-weight: bold;">test_data_summary</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box; color: rgb(153, 153, 153);">self</span><span style="box-sizing: border-box;">):</span>
    <span style="box-sizing: border-box;">expected_df</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> <span style="box-sizing: border-box; color: rgb(153, 153, 153);">self</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">exp_df</span>
    <span style="box-sizing: border-box;">result_df</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> <span style="box-sizing: border-box;">data_summary</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box; color: rgb(153, 153, 153);">self</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">df</span><span style="box-sizing: border-box;">)</span>
    <span style="box-sizing: border-box; color: rgb(153, 153, 153);">self</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">assertTrue</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">expected_df</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">equals</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">result_df</span><span style="box-sizing: border-box;">))</span>

<span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">def</span> <span style="box-sizing: border-box; color: rgb(153, 0, 0); font-weight: bold;">test_display</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box; color: rgb(153, 153, 153);">self</span><span style="box-sizing: border-box;">):</span>
    <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">print</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box; color: rgb(221, 17, 68);">'---- Data summary ----'</span><span style="box-sizing: border-box;">,</span> <span style="box-sizing: border-box; color: rgb(153, 153, 153);">self</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">exp_df</span><span style="box-sizing: border-box;">,</span> <span style="box-sizing: border-box;">sep</span><span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span><span style="box-sizing: border-box; color: rgb(221, 17, 68);">'</span><span style="box-sizing: border-box; color: rgb(221, 17, 68);">\n</span><span style="box-sizing: border-box; color: rgb(221, 17, 68);">'</span><span style="box-sizing: border-box;">)</span>
    <span style="box-sizing: border-box;">expected_stdout</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> <span style="box-sizing: border-box; color: rgb(153, 153, 153);">self</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">capsys</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">readouterr</span><span style="box-sizing: border-box;">()</span>
    <span style="box-sizing: border-box;">display_summary</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box; color: rgb(153, 153, 153);">self</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">df</span><span style="box-sizing: border-box;">)</span>
    <span style="box-sizing: border-box;">result_stdout</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> <span style="box-sizing: border-box; color: rgb(153, 153, 153);">self</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">capsys</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">readouterr</span><span style="box-sizing: border-box;">()</span>
    <span style="box-sizing: border-box; color: rgb(153, 153, 153);">self</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">assertEqual</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">expected_stdout</span><span style="box-sizing: border-box;">,</span> <span style="box-sizing: border-box;">result_stdout</span><span style="box-sizing: border-box;">)</span>

if name == 'main':
unittest.main()

Обратите внимание на декоратор @pytest.fixture(autouse=True) и функцию, которую он оборачивает (_pass_fixture).  В терминологии модульного тестирования этот метод называется фикстурой (fixture). Фикстуры - это функции (или методы, если Вы используете подход ООП), которые будут выполняться перед каждым тестом, к которому они применяются. Фикстуры используются для передачи данных в тесты. Они выполняют ту же задачу, что и метод setUp(), который мы использовали ранее. Здесь мы используем заранее определенное фикстуру под названием capsys для захвата стандартного вывода (stdout) и повторного использования его в нашем тесте. Теперь изменим соответствующим образом функцию display_summary():

import pandas as pd
def data_summary(df: pd.DataFrame) -> pd.DataFrame:
"""
Function defined to output details about the number
of rows and columns and the column dtype frequency of
the passed pandas DataFrame
"""
def _shape(df: pd.DataFrame) -> pd.DataFrame:
"""
Function defined to return a dataframe with details about
the number of row and columns
"""
row, col = df.shape
return pd.DataFrame(data=[[row], [col]], columns=['Values'], index=['Number of rows', 'Number of columns'])
<span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">def</span> <span style="box-sizing: border-box; color: rgb(153, 0, 0); font-weight: bold;">_dtypes_freq</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">df</span><span style="box-sizing: border-box;">:</span> <span style="box-sizing: border-box;">pd</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">DataFrame</span><span style="box-sizing: border-box;">)</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">-&gt;</span> <span style="box-sizing: border-box;">pd</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">DataFrame</span><span style="box-sizing: border-box;">:</span>
    <span style="box-sizing: border-box; color: rgb(221, 17, 68);">"""
    Function defined to return a dataframe with details about
    the pandas dtypes frequency
    """</span>
    <span style="box-sizing: border-box;">counter</span><span style="box-sizing: border-box;">,</span> <span style="box-sizing: border-box;">types</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> <span style="box-sizing: border-box;">{},</span> <span style="box-sizing: border-box;">df</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">dtypes</span>
    <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">for</span> <span style="box-sizing: border-box;">dtype</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">in</span> <span style="box-sizing: border-box;">types</span><span style="box-sizing: border-box;">:</span>
        <span style="box-sizing: border-box;">tmp</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> <span style="box-sizing: border-box; color: rgb(0, 134, 179);">str</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">dtype</span><span style="box-sizing: border-box;">)</span>
        <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">if</span> <span style="box-sizing: border-box;">tmp</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">in</span> <span style="box-sizing: border-box;">counter</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">keys</span><span style="box-sizing: border-box;">():</span>
            <span style="box-sizing: border-box;">counter</span><span style="box-sizing: border-box;">[</span><span style="box-sizing: border-box;">tmp</span><span style="box-sizing: border-box;">]</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">+=</span> <span style="box-sizing: border-box; color: rgb(0, 153, 153);">1</span>
        <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">else</span><span style="box-sizing: border-box;">:</span>
            <span style="box-sizing: border-box;">counter</span><span style="box-sizing: border-box;">[</span><span style="box-sizing: border-box;">tmp</span><span style="box-sizing: border-box;">]</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> <span style="box-sizing: border-box; color: rgb(0, 153, 153);">1</span>
    <span style="box-sizing: border-box;">values</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> <span style="box-sizing: border-box;">[[</span><span style="box-sizing: border-box;">value</span><span style="box-sizing: border-box;">]</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">for</span> <span style="box-sizing: border-box;">value</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">in</span> <span style="box-sizing: border-box;">counter</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">values</span><span style="box-sizing: border-box;">()]</span>
    <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">return</span> <span style="box-sizing: border-box;">pd</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">DataFrame</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">data</span><span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span><span style="box-sizing: border-box;">values</span><span style="box-sizing: border-box;">,</span> <span style="box-sizing: border-box;">columns</span><span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span><span style="box-sizing: border-box;">[</span><span style="box-sizing: border-box; color: rgb(221, 17, 68);">'Values'</span><span style="box-sizing: border-box;">],</span> <span style="box-sizing: border-box;">index</span><span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span><span style="box-sizing: border-box; color: rgb(0, 134, 179);">list</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">counter</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">keys</span><span style="box-sizing: border-box;">()))</span>
<span style="box-sizing: border-box;">result_df</span> <span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">=</span> <span style="box-sizing: border-box;">pd</span><span style="box-sizing: border-box;">.</span><span style="box-sizing: border-box;">concat</span><span style="box-sizing: border-box;">([</span><span style="box-sizing: border-box;">_shape</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">df</span><span style="box-sizing: border-box;">),</span> <span style="box-sizing: border-box;">_dtypes_freq</span><span style="box-sizing: border-box;">(</span><span style="box-sizing: border-box;">df</span><span style="box-sizing: border-box;">)])</span>
<span style="box-sizing: border-box; color: rgb(0, 0, 0); font-weight: bold;">return</span> <span style="box-sizing: border-box;">result_df</span>

def display_summary(df: pd.DataFrame) -> None:
"""
Function define to print out the result of the data summary
"""
result_df = data_summary(df)
message = '---- Data summary ----'
print(message, result_df, sep='\n')

Ещё раз запустим тесты:

poetry run pytest -v

==================================================== test session starts ====================================================
platform darwin -- Python 3.8.7, pytest-5.4.3, py-1.11.0, pluggy-0.13.1 -- /Users/aabur/Documents/GitHub/AABur/modern_python/summarize_dataframe/.venv/bin/python
cachedir: .pytest_cache
rootdir: /Users/aabur/Documents/GitHub/AABur/modern_python/summarize_dataframe
collected 2 items
tests/test_summarize_dataframe.py::TestDataSummary::test_data_summary PASSED                                          [ 50%]
tests/test_summarize_dataframe.py::TestDataSummary::test_display PASSED                                               [100%]
===================================================== 2 passed in 0.31s =====================================================

Тесты прошли успешно. Пришло время зафиксировать нашу работу и поделиться ею, например, опубликовав на GitHub. Перед этим давайте подробно рассмотрим, как правильно сообщать о нашей работе с помощью сообщений о коммитах Git, соблюдая и поддерживая единый стандарт.

Применение правил по созданию сообщений Git-коммитов в проекте Python

Написание оптимальных Git-коммит сообщений - непростая задача. Сообщения должны быть четкими, читаемыми и понятными в долгосрочной перспективе. Спецификация Conventional Commits предлагает набор правил для создания однозначной истории коммитов. На Хекслете есть большая статья, посвященная правильному именованию коммитов.

Использование commitizen

Мы будем использовать пакет commitizen для интеграции Conventional Commits в наш проект на Python. Добавим этот пакет в зависимости разработчика:

poetry add -D commitizen

Чтобы настроить commitizen для своего проекта, выполните команду cz init. Она предложит нам ответить на ряд вопросов:

cz init

? Please choose a supported config file: (default: pyproject.toml) (Use arrow keys)
» pyproject.toml
  .cz.toml
  .cz.json
  cz.json
  .cz.yaml
  cz.yaml
  
? Please choose a cz (commit rule): (default: cz_conventional_commits) (Use arrow keys)
» cz_conventional_commits
  cz_jira
  cz_customize
  
? Please enter the correct version format: (default: "$version")

? Do you want to install pre-commit hook? (Y/n)

Выберем здесь все варианты по умолчанию, так как они полностью соответствуют нашей реальной ситуации. Последний вопрос спрашивает нас, нужно ли использовать хук pre-commit. Мы собираемся вернуться к этому позже. Поэтому пока просто ответим «нет»(n). Если мы посмотрим на файл pyproject.toml, то увидим, что была добавлена новая запись под названием [tool.commitizen]:

[tool.commitizen]
name = "cz_conventional_commits" # правило формирования коммит-сообщений
version = "0.0.1"
tag_format = "$version"

Проверить коммит-сообщение, можно при помощи следующей команды:

cz check -m "all summarize_data tests now succeed"

commit validation: failed!
please enter a commit message in the commitizen format.
commit "": "all summarize_data tests now succeed"
pattern: (build|ci|docs|feat|fix|perf|refactor|style|test|chore|revert|bump)!?((\S+))?:(\s.*)

Наше сообщение отклонено, потому что оно не соответствует выбранным правилам для коммит-сообщений. Последняя строка предлагает некоторые шаблоны для использования. Уделите немного времени чтению документации о соглашении о коммитах и выполните команду cz info, чтобы распечатать краткую документацию:

cz info

The commit contains the following structural elements, to communicate
intent to the consumers of your library:

fix: a commit of the type fix patches a bug in your codebase
(this correlates with PATCH in semantic versioning).

feat: a commit of the type feat introduces a new feature to the codebase
(this correlates with MINOR in semantic versioning).

BREAKING CHANGE: a commit that has the text BREAKING CHANGE: at the beginning of
its optional body or footer section introduces a breaking API change
(correlating with MAJOR in semantic versioning).
A BREAKING CHANGE can be part of commits of any type.

Others: commit types other than fix: and feat: are allowed,
like chore:, docs:, style:, refactor:, perf:, test:, and others.
[...]

Эта команда подскажет вам, как написать сообщение о коммите. Здесь формат должен быть таким: «[тип]: [СООБЩЕНИЕ]». Для нас это выглядит так:

cz check -m "test: all summarize_data tests now succeed"

Commit validation: successful!

Очень хорошо, наше коммит-сообщение считается корректным. Но подождите. Проверять коммит-сообщения каждый раз с помощью commitizen может быть утомительно, и это не даёт гарантии, что коммит будет принят. Было бы лучше автоматически проверять сообщение каждый раз, когда мы используем команду git commit. Именно в этом случае в действие вступает pre-commit хук.

Автоматическое соблюдение соглашений о Git-коммитах при помощи pre-commit

Хуки Git полезны для автоматизации и выполнения некоторых действий на разных этапах жизненного цикла Git. Хук pre-commit позволяет запускать скрипты до того, как будет выполнен Git-коммит. Мы можем использовать хук для проверки сообщений о коммитах и предотвращения использования Git сообщения, которое не соответствует нашим ожиданиям. Хук активен как из командной строки, так и из любых инструментов, взаимодействующих с репозиторием Git, в котором зарегистрирован хук, включая вашу любимую IDE.

pre-commit — это фреймворк для управления и поддержки многоязычных хуков pre-commit. Если вы хотите узнать больше о внутреннем устройстве и спектре возможностей pre-commit, то можете прочитать документацию по его использованию.

Чтобы установить pre-commit, просто выполните команду:

poetry add -D pre-commit

Для автоматизации проверки коммита Git нам сначала нужно создать конфигурационный файл .pre-commit-config.yaml:

---
repos:
  - repo: https://github.com/commitizen-tools/commitizen
    rev: master
    hooks:
      - id: commitizen
        stages: [commit-msg]

Далее мы можем установить хук с источником, определенным в параметре repo:

pre-commit install --hook-type commit-msg
pre-commit installed at .git/hooks/commit-msg

Теперь, когда все готово, мы можем использовать наш Git-хук:

git add tests/test_summarize_dataframe.py

git commit -m "test: all summarize_data tests now succeed"

[WARNING] Unstaged files detected.
[INFO] Stashing unstaged files to /Users/aabur/.cache/pre-commit/patch1637958717.
commitizen check.........................................................Passed
[INFO] Initializing environment for https://github.com/commitizen-tools/commitizen.
[INFO] Installing environment for https://github.com/commitizen-tools/commitizen.
[INFO] Once installed this environment will be reused.
[INFO] This may take a few minutes...
commitizen check.........................................................Passed
[INFO] Restored changes from /Users/aabur/.cache/pre-commit/patch1637958717.
[main 6ef0700] test: all summarize_data tests now succeed
1 file changed, 38 insertions(+), 5 deletions(-)
rewrite tests/test_summarize_dataframe.py (98%)

pre-commit устанавливает среду для выполнения своих проверок. Как вы можете видеть здесь, сообщение о коммите прошло проверку. В завершение мы можем закоммитить изменения, внесенные в файлы сборки (poetry.lockpyproject.toml) и наш модуль:

git add poetry.lock pyproject.toml

git commit -m "build: add developer dependencies"

commitizen check.........................................................Passed
[main 8e616bc] build: add developer dependencies
2 files changed, 664 insertions(+), 3 deletions(-)

git add .pre-commit-config.yaml

git commit -m "build: add pre-commit hook"

commitizen check.........................................................Passed
[main 60880cb] build: add pre-commit hook
1 file changed, 7 insertions(+)

git add summarize_dataframe/summarize_df.py

git commit -m "feat: implementation of the summary function to summarize dataframe"

commitizen check.........................................................Passed
[main 53a82a0] feat: implementation of the summary function to summarize dataframe
1 file changed, 42 insertions(+)

Теперь мы можем отправить все в наш репозиторий GitHub:

git push origin main

Enumerating objects: 18, done.
Counting objects: 100% (18/18), done.
Delta compression using up to 12 threads
Compressing objects: 100% (12/12), done.
Writing objects: 100% (12/12), 19.60 KiB | 6.53 MiB/s, done.
Total 12 (delta 3), reused 0 (delta 0), pack-reused 0
remote: Resolving deltas: 100% (3/3), completed with 2 local objects.
To github.com:AABur/summarize_dataframe.git
af38079..53a82a0  main -&gt; main

Заключение

Мы рассмотрели несколько тем:

  • На первом этапе мы узнали, как писать модульные тесты для вашего кода. Писать тесты до кода поможет вам уточнить API и ожидаемый результат до реализации в коде. Мы использовали unittest, который уже доступен в стандартной библиотеке Python. Также было продемонстрировано использование библиотеки pytest для написания тестов. Очень удобно то, что pytest с самого начала поддерживает класс unittest.TestCase. Вы можете писать свои тесты с помощью любой из двух библиотек или даже смешивать их в зависимости от ваших потребностей и иметь одну общую команду для запуска всех тестов.

  • Мы рассмотрели, как обеспечить соблюдение хороших практик при написании сообщений о коммитах в Git. Предлагаемое нами решение основано на использовании двух пакетов Python: commitizen и pre-commit. Первый предоставляет инструменты для проверки соответствия сообщения выбранным вами соглашениям. Второй автоматизирует процесс с помощью Git-хука.

Краткая памятка

poetry

  • Добавьте зависимости проекта

poetry add [package_name]

  • Установить глобальную версию Python

pyenv global

  • Установить локальную версию Python

pyenv local

poetry

  • Добавьте зависимости проекта

    poetry add [package_name]

  • Добавьте зависимости для разработки

    poetry add -D [package_name]

  • Запуск тестов

    poetry run pytest

commitizen

  • Инициализация commitizen

    cz init

  • Проверка коммита

    cz check -m "YOUR MESSAGE"

pre-commit

  • Создание файла конфигурации по умолчанию

    pre-commit sample-config

  • Установить git-хук

    pre-commit install --hook-type [hook_name]

Источник: https://habr.com/ru/post/645721/

Интересные статьи

Интересные статьи

В первой части серии мы разобрали, что такое первичные и вторичные космические лучи, рассмотрели способ, которым будем их регистрировать и познакомились с проектом CosmicWatch. В пре...
Хочу поделиться опытом автоматизации экспорта заказов из Aliexpress в несколько CRM. Приведенные примеры написаны на PHP, но библиотеки для работы с Aliexpress есть и для...
В статье описан способ, который упрощает работу над созданием скетчей. Он подойдет всем людям, работа которых — проектировать интерфейсы для пользователей. Читать дальше → ...
В 2019 году люди знакомятся с брендом, выбирают и, что самое главное, ПОКУПАЮТ через интернет. Сегодня практически у любого бизнеса есть свой сайт — от личных блогов, зарабатывающих на рекламе, до инт...
Как обновить ядро 1С-Битрикс без единой секунды простоя и с гарантией работоспособности платформы? Если вы не можете закрыть сайт на техобслуживание, и не хотите экстренно разворачивать сайт из бэкапа...