Python Unit Test Automation

Listing 3-10.  test_me.py
def add(x, y):
    return(x + y)
def mul(x, y):
    return(x * y)
def sub(x, y):
    return(x - y)
def div(x, y):
    return(x / y)
Now, since test_me.py is in the test directory, it can directly be imported into 
another module in the same directory using the import test_me statement. The test 
module in Listing 
3-11
 imports test_me.py to test its functionality.
Listing 3-11.  test_module08.py
import unittest
import test_me
class TestClass09(unittest.TestCase):
        def test_case01(self):
                self.assertEqual(test_me.add(2, 3), 5)
                print("\nIn test_case01()")
        def test_case02(self):
                self.assertEqual(test_me.mul(2, 3), 6)
                print("\nIn test_case02()")

---

Chapter 3 
■
 Unittest
49
Run the test module with the following command:
python3 -m unittest -v test_module08
The output is as follows:
test_case01 (test_module08.TestClass09) ...
In test_case01()
ok
test_case02 (test_module08.TestClass09) ...
In test_case02()
ok
----------------------------------------------------------------------
Ran 2 tests in 0.004s
OK
This way, you can organize the development code and the testing code in the same 
directory, in the different files.
Placing the Development and Test Code in Separate Directories
Many coding standards recommend that the development code and the testing code files 
be organized in separate directories. Let’s do that now.
Navigate to the chapter03 directory,
cd /home/pi/book/code/chapter03
Create a new directory called mypackage in the chapter03 directory:
mkdir mypackage
Navigate to the mypackage directory:
cd mypackage
Save the code in Listing 
3-12
 as the file mymathlib.py in the mypackage directory,