< Unit Testing < Pytest

Create a Python Program

Create the following Python program. Save it as multiply.py

def multiply(x, y):
    result = x * y
    return result

Create a Test Program

Create the following Pytest test program. Save it as either test_multiply.py or multiply_test.py

import unittest
from multiply import *

class MultiplyTestCase(unittest.TestCase):
    def test_multiply(self):
        self.assertEqual(
            multiply(2, 2), 4, "Error: 2 * 2 = 4"
        )

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

Test Success

Test the program by running pytest in a terminal or command prompt in the same folder as the two programs above and observe the results.

pytest

Test Failure

Change the multiply source code somehow, such as multiplying by 0 rather than multiplying by y. Test the program by running pytest again and observe the results.

This article is issued from Wikiversity. The text is licensed under Creative Commons - Attribution - Sharealike. Additional terms may apply for the media files.