import unittest
from name_function import get_formatted_name


class NamesTestCase(unittest.TestCase):
    """test name_function.py"""

    def test_first_last_name(self):
        formatted_name = get_formatted_name('lin', 'du')
        self.assertEqual(formatted_name, '<NAME>')

    def test_first_last_middle_name(self):
        formatted_name = get_formatted_name('mr', 'lin', 'du')
        self.assertEqual(formatted_name, '<NAME>')


unittest.main()

import unittest
from name_function import get_formatted_name


class NamesTestCase(unittest.TestCase):
    """test name_function.py"""

    def test_first_last_name(self):
        formatted_name = get_formatted_name('lin', 'du')
        self.assertEqual(formatted_name, '<NAME>')

    def test_first_last_middle_name(self):
        formatted_name = get_formatted_name('mr', 'lin', 'du')
        self.assertEqual(formatted_name, '<NAME>')


unittest.main()

test name_function.py