Функция AssertEqual() модуля unittest в Python

• Что такое функция AssertEqual() в Python?
• Пример с AssertEqual() в Python
• Метод AssertNotEqual() в Python

Что такое функция AssertEqual() в Python?

AssertEqual() — это метод класса TestCase модуля unittest в Python. Функция AssertEqual() проверяет, равны ли два значения:

assertEqual(first, second, msg=None)

Если первое значение не равно второму значению, тест завершится неудачей. msg не является обязательным. Если msg предоставлено, оно будет показано в результатах теста, если тест не пройден.

Пример с AssertEqual() в Python

• Сначала создайте новый модуль с именем main.py и определите функцию add():

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

• Во-вторых, создайте тестовый модуль test_main.py для проверки функции add():

import unittest

from main import add


class TestMain(unittest.TestCase):
    def test_add(self):
        self.assertEqual(add(1, 2), 3)

• В-третьих, запустите тест:

python -m unittest test_main.py -v

Выход:

test_add(test_main.TestMain) ... ok

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

OK

Метод AssertNotEqual() в Python

Метод AssertNotEqual() проверяет, не равны ли два значения:

assertNotEqual(first, second, msg=None)

Если первое равно второму, тест не пройден. Например:

import unittest

from main import add


class TestMain(unittest.TestCase):
    def test_add(self):
        self.assertEqual(add(1, 2), 3)

    def test_add_floats(self):
        self.assertNotEqual(add(0.2, 0.1), 0.3)

Запустите тест:

python -m unittest test_main.py -v

Выход:

test_add(test_main.TestMain) ... ok
test_add_floats(test_main.TestMain) ... ok

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

OK

Поскольку 0,2 + 0,1 возвращает 0,30000000000000004, оно не равно 0,3. Таким образом, следующий тест пройден:

self.assertNotEqual(add(0.2, 0.1), 0.