Using PyUnit For Testing A Selenium Python Test Suite
Tags: python, beginners, webdev, testing
Published: 2019-01-30 09:02:10

---
title: Using PyUnit For Testing A Selenium Python Test Suite
published: true
description: PyUnit is a unit testing framework which is derived from JUnit for compatible execution of Selenium with Python language. Here is how you can leverage PyUnit. 
tags: #python #beginners #webdev #testing
cover_image: https://www.lambdatest.com/blog/wp-content/uploads/2019/01/20190107_182900_0000-01-1.jpeg
canonical_url: https://www.lambdatest.com/blog/using-pyunit-for-testing-a-selenium-python-test-suite/
---

Selenium, as we know, is an open-source test suite used for cross-platform browser automation. Due to its popularity and ease of use, many compatible test frameworks with respect to different programming languages have been developed to extend the support of [Selenium](https://www.lambdatest.com/blog/selenium-webdriver-tutorial-for-cross-browser-testing/?utm_source=devto&utm_medium=organic&utm_campaign=jul13_rd&utm_term=rd&utm_content=blog) for [cross-browser testing](https://www.lambdatest.com/?utm_source=devto&utm_medium=organic&utm_campaign=jul13_rd&utm_term=rd&utm_content=webpage) over multiple programming languages. Today we’re going to look into the Selenium Python tutorial to run our first automation script using the PyUnit framework with Selenium.

PyUnit is a unit testing framework that is derived from JUnit for compatible execution of Selenium with Python language. PyUnit is a very popular testing framework to perform ‘Unit Testing’ — a mechanism to validate a particular piece of functionality in a module. The popularity of PyUnit as a Python-specific unit testing framework has been a major reason for including PyUnit as an official Python module from version 2.5. The Python UnitTest library, referred to as unittest, is widely used to perform [Selenium Python testing](https://www.lambdatest.com/selenium-python-testing?utm_source=devto&utm_medium=organic&utm_campaign=jul13_rd&utm_term=rd&utm_content=webpage). We will leverage the unittest library in this Selenium Python tutorial.

## Building Blocks Of Python Unit Test [PyUnit] Framework

Let’s start the Selenium Python tutorial by understanding the building blocks of the PyUnit framework. The unittest framework contains the following core modules that are core to test case development & execution:

* **Test Loader**: The Test Loader Python class is used to load test cases and test suites. It is also termed ‘Test Fixture’. Test Cases and Test Suite can be created locally or can be loaded from an external file/database. On successful execution, the Test Suite object is released by the Test Loader, and the same object can be used throughout the execution of the Test Case.

* **Test Runner**: Responsible for displaying the output of the executed test to an end-user using a runnable interface. The output could be represented in a variety of ways. It could be displayed as a standard code in the form of a GUI or through a textual representation.

* **Test Suite**: Test Suite is a group of test cases that are ‘grouped logically’ based on the functionalities that are under test. Test suites are used to make the test code more modular & maintainable.

* **Test Case**: The test case contains the actual implementation of the test code. In case some functionality is not ready, a developer can make use of Stubs i.e. dummy source code, to test source code at a unit level.

* **Test Report**: Test Reports are ideally used to organize the results of test case execution i.e. whether the test case execution result is pass/fail. It also details the time taken for execution, execution summary, etc. The better the maintenance of the test report, the easier it extracts useful information from it.

## PyUnit/unittest Overview

PyUnit is also referred to as unittest and works in a similar manner as xUnit, which is a very popular unit testing framework whose structure & functionality is derived from [Smalltalk’s SUnit](http://sunit.sourceforge.net/). Hence, a developer who has prior experience with xUnit or other popular unit testing frameworks would find unittest easier to understand.

Testcase/Testsuite written using unittest module follows a common format

 1. Define a class derived from unittest.TestCase(). We discuss TestCase() in more detail in subsequent sections.

 2. Define test cases having the nomenclature test_name

 3. Execute the testcases/testsuites by having unittest.main(), which is normally placed at the bottom of the file i.e. after the necessary classes for testcase execution have been implemented.

We will have a look at the important classes in the unittest package in further sections.

***Did you know? [LT Debug Chrome Extension](https://www.lambdatest.com/lt-debug?utm_source=devto&utm_medium=organic&utm_campaign=jul13_rd&utm_term=rd&utm_content=lt_debug) makes debugging any web page a breeze.***

## setUp() and tearDown() — Initialization & De-Initialization

The setUp() method is an entry point to the test cases. It does not have any arguments. Since it is the entry point, some of the important initialization steps are performed in the implementation of the setUp() method. Below are some things related to initialization that can be included in setUp()

* Load the test data.

* Creation of a browser instance e.g. Firefox using the WebDriver interface.

* Opening files for I/O (Input/Output) operation, either for reading the test data or appending the results of the test execution.

Below is an example of the initialization & cleanup activity performed via setUp() and tearDown() methods

    import unittest
    #Import other modules that are required for testing

    class SearchText(unittest.TestCase):
     def setUp(self):
      #create a new FireFox session
      self.driver = webdriver.FireFox()
      .................................
      .................................
      self.driver.get("[https://www.lambdatest.com](https://www.lambdatest.com)")
      
     def tearDown(self):
      .................................
      .................................
      self.driver.quit()

In the above example, an instance of Firefox browser is created using the WebDriver API (Line 7). On successful creation, the Home Page [https://www.lambdatest.com](https://www.lambdatest.com/) is opened on the Firefox browser. All the necessary operations which are required for unit testing are then performed. Once the unit test is complete, the cleanup is performed in the tearDown() API (Line 12). In a nutshell, setUp() & tearDown() are executed before & after each every test method.

Now that you have understood the basics of the initialization & de-initialization of a Unit test case written in Python, let’s have a look at some of the important classes in this selenium python tutorial.

## PyUnit — Classes and Functions

**class unittest.TestCase(methodName=’TestName’)**

We will have a look at the important classes in the unittest package in further sections of this Selenium Python tutorial. Specific test cases are implemented in subclasses. In many scenarios, no changes would be required in methodName nor runTest() needs to be implemented. We have already discussed the important initialization methods setUp() and tearDown(), and now we will have a look at the other methods.

* **setUpClass()**

* A class method is called before any of the individual tests are called. @classmethod is the identifier with which you can identify a setUpClass(). setUpClass() has only one argument i.e. the class name.

* 

* **tearDownClass()**

* This method is called after all the tests in the class are executed. Similar to setUpClass(), tearDownClass() also as only one argument i.e. the class name. The ‘class name’ should match with the name which is used in the setUpClass(), else it might result in an error. A sample implementation citing the usage of setUpClass() and tearDownClass() is shown below:

    import unittest

    class Example(unittest.TestCase):
        [@classmethod](http://twitter.com/classmethod)
        def setUpClass(class_name):
            print("setUpClass executed")

    def setUp(self):
            print("setUp executed")

    def test_1(self):
            print("test-1 executed")

    def test_2(self):
            print("test-2 executed")

    def tearDown(self):
            print("tearDown executed")

    [@classmethod](http://twitter.com/classmethod)
        def tearDownClass(class_name):
            print("tearDownClass executed")

One question that could arise — What is the basic difference between setUp() and setUpClass() [or tearDown() & tearDownClass()]? setUp() [and its de-initialization counterpart tearDown()] are loaded & executed before each test method, whereas setUpClass() [and its de-initialization counterpart tearDownClass()] are executed once for the whole class. Let’s have a look at a simple example to understand the difference.

As seen in the below example [SetupClass-Python.py], depending on the test case being executed [test_1 or test_2], setUpClass gets executed once in the beginning, and then the corresponding test case is executed. After the execution of the test case, the code in tearDownClass() is executed.

In order to execute the code, press CTRL+F9. As seen in the snapshot below, both the test cases observed in this Selenium Python tutorial were executed in the below sequence of execution:

* Implementation under setUpClass() is executed

* test_1 is executed (setUp & tearDown is called for test_1 i.e. initialization & cleanup after execution of test_1)

* test_2 is executed (setUp & tearDown is called for test_2 i.e. initialization & cleanup after execution of test_2)

* Implementation under tearDownClass() is executed

As explained earlier, setUpClass() & tearDownClass() are executed only once, whereas setUp() & tearDown() are executed for each test method.

![](https://cdn-images-1.medium.com/max/2118/1*0uX133L8QYpPHc2LZ11Mgg.png)

* **run(result=None)**:

TestResult object is passed as a parameter to the run() method. The parameter is optional, and if the object is not supplied to the method, a temporary object is created & used to store the results. The result is passed to the caller of the run() method. We will discuss the TestResult object in detail in a subsequent section.

***Did you know? [Content Security Policy](https://www.lambdatest.com/lt-debug/content-security-policy?utm_source=devto&utm_medium=organic&utm_campaign=jul13_rd&utm_term=rd&utm_content=lt_debug) on LT Debug Chrome Extension helps generate Content Security Policy headers on any website or web page of your choice.***

## class unittest.TestSuite(tests=())

The TestSuite class is an aggregation of individual testcases & test suites. Instead of executing testcases on an iterative basis, developers & testers can make use of this class since makes code maintenance easy.

Unlike the TestCase() class, the TestSuite() class does not contain any implementation of test code since it is used to group testcases on a logical/functional basis. Some of the methods that can be used along with TestSuite() class are mentioned below

* **addTest(test)**: It is used to add TestCase or TestSuite

* **addTests(tests)**: In case you plan to iterate through the testcases or testsuites, you can make use of addTests().

* **debug()**: This method is primarily used for debugging purpose since it does not return the execution results.

* **run(result)**: In case you plan to execute testcases associated with a testsuite, you can make use of run(result). The result object stores the execution results.

## class unittest.TestResult

TestResult class is used to gather information about the number of test cases that have passed/failed/skipped. As discussed earlier, the TestCase & TestSuite classes are used to record the results of the testcase/testsuite execution. Some of the important attributes of TestResult are failures, errors, skipped, expectedFailures, unexpectedSuccesses.

***Here’s an article on the [30 Top Automation Testing Tools In 2022](https://www.lambdatest.com/blog/automation-testing-tools/?utm_source=devto&utm_medium=organic&utm_campaign=jul13_rd&utm_term=rd&utm_content=lt_debug)!***

## Assertions In PyUnit Test Framework

Assertions are very popular in testing, irrespective of the programming language being used for test-code development. An assertion is nothing but a Boolean expression (representing ‘True/False’) which would carry a value ‘True’ till the time it [encounters a bug/issue](https://www.lambdatest.com/blog/27-tips-for-finding-bugs-in-your-website/?utm_source=devto&utm_medium=organic&utm_campaign=jul13_rd&utm_term=rd&utm_content=blog) in the test code. Three types of assertions are available to check equivalence, comparison, and performing actions in case of exceptions.

Next in this Selenium Python tutorial, we will have a look at some of the important assertions for PyUnit framework.

![](https://cdn-images-1.medium.com/max/2000/1*c416IAaGZO2vYcSaWMhQpQ.png)

With the basics of PyUnit/unittest covered, let’s have a look at a sample code that can delve into the aspects which have been discussed so far

    '''This sample program demonstrates the PyUnit/unittest framework'''

    # Inclusion of the PyUnit Test Framework
    import unittest
      
    def addition(x,y,z=0):
       # Add the three parameters which are passed to the addition function 
       return x+y+z
     
    # This is where the test case is implemented
    class AddTest(unittest.TestCase):
       def setUp(self):
          # initialization code for the testcase/testsuite can be added here
          pass
     
       # These are tests that should be performed once the basic premise is set
       # Unit Tests start with test_
       # The addition method which was implemented earlier will be used here
       
       def test_addtion(self):
          self.assertEqual(addition(10,11,12), 33)
          # x=11, y=12, z=44 if (x+y)=z, the test would raise an assert since
          # the test is for assertNotEqual operation
          self.assertNotEqual(addition(11,12), 44)
     
       def test_negative_values(self):
          self.assertEqual(addition(-9,25), 16)
          self.assertNotEqual(addition(-9,25), 17)      
     
       def tearDown(self):
          # Deinit and cleanup should be done here
          pass   
     
    if __name__=='__main__':
       unittest.main()

The above code comprises of two testcases — test_addition() and test_negative_values(). A method named addition() [Line 6~ Line 8] is created to add three parameters that are passed to that method. assertNotEqual and assertEqual are used in the testcases. Respective Testcases would pass if the conditions in the Assert equate to TRUE. When the above code is executed, the final test result is PASS as conditions mentioned in assertEqual and assertEqual are fulfilled e.g. In test_addition() testcase, assert would be invoked if the addition of x=10,y=11,z=12 does not equate to 33 since it is testing for ‘Equate’ operation. Below is the screenshot of the output when the code is compiled & tested from the command prompt [file — Python-unittest-output1.png]

![](https://cdn-images-1.medium.com/max/2000/1*-MsK4oRc4j6Z08HyXu8wyQ.png)

In order to invoke an Assert, we make a small change in the test_negative_values() testcase. Below is the modified source code [changes are marked as comment]

    '''This sample program demonstrates the PyUnit/unittest framework'''

    # Inclusion of the PyUnit Test Framework
    import unittest
      
    def addition(x,y,z=0):
       # Add the three parameters which are passed to the addition function 
       return x+y+z
     
    # This is where the test case is implemented
    class AddTest(unittest.TestCase):
       def setUp(self):
          # initialization code for the testcase/testsuite can be added here
          pass
     
       # These are tests that should be performed once the basic premise is set
       # Unit Tests start with test_
       # The addition method which was implemented earlier will be used here
       
       def test_addtion(self):
          self.assertEqual(addition(10,11,12), 33)
          # x=11, y=12, z=44 if (x+y)=z, the test would raise an assert since
          # the test is for assertNotEqual operation
          self.assertNotEqual(addition(11,12), 44)
     
       def test_negative_values(self):
          self.assertNotEqual(addition(-9,25), 16)   #Changed Values#   
     
       def tearDown(self):
          # Deinit and cleanup should be done here
          pass   
     
    if __name__=='__main__':
       unittest.main()

As seen on Line-27, assert would be issued if (x+y)!=z. In our example, x=-9, y=25, and z=16. As the test condition fails, it issues to assert. As you can see below in this Selenium Python Tutorial, the snapshot of the output.

![](https://cdn-images-1.medium.com/max/2000/1*PN_7b88Nx8sF-YLkyk_Qmg.png)

## Summary:

This Selenium Python tutorial helped us to understand some of the important aspects of PyUnit. Especially how this effective Unit testing framework can be used to automate cross-browser testing code and how ‘asserts’ can be used effectively while testing your code. PyUnit can also be used to perform module-level testing since you also have the flexibility to execute corner test cases as well. During the course of the article, we used Notepad++ & ‘command line Python’ for writing PyUnit code & execution. As a developer, you should use the IDE & programming environment of your choice & comfort. Though PyUnit was written for ‘C’ Python, it is now possible to write PyUnit tests for languages like Java using the Jython framework. Covering Jython is beyond the scope of this article. You can find the PyUnit/unittest official documentation [here](https://docs.python.org/3/library/unittest.html).