allow usage of unittest's load_tests protocol in python unit tests

[benjaminbecker]

I have a project with the following structure:

.
├── functions.py
├── __init__.py
└── tests
    ├── __init__.py
    ├── test_functions.py

This is the content of test_functions.py:

import unittest
import os.path as path

from functions import sum


current_dir = path.dirname(path.abspath(__file__))
test_file = path.join(current_dir, 'foo.txt')


class TestFunctions(unittest.TestCase):

    def test_sum(self):
        self.assertEqual(sum(1, 2), 3)

    def test_file_exists(self):
        self.assertTrue(path.exists(test_file))

In order to make the second test succeed I need to run some code before it. I do this by using unittest's load_test protocol. It is implemented in tests/__init__.py:

import os
import unittest
from .test_functions import TestFunctions, test_file


class CustomTestSuite(unittest.TestSuite):

    def run(self, result):
        with open(test_file, "w") as file:
            file.write("")
        super().run(result)
        os.remove(test_file)



def load_tests(loader, default_tests, pattern):
    # top level directory cached on loader instance
    test_suite = CustomTestSuite()
    tests = loader.loadTestsFromTestCase(TestFunctions)
    test_suite.addTests(tests)
    return test_suite

If I run python -m unittest from the root dir, everything works fine:

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

OK

I could not manage to use my custom test suite within the vs code python extension. If there is currently no solution, I would like to request this feature.

I am aware of other solutions like using setUp per test or module for this simple example. However in my real project the setup involves more complex steps like initializing an email server.

[karthiknadig]

Thank you for the suggestion! We have marked this issue as "needs decision" to make sure we have a conversation about your idea. We plan to leave this feature request open for at least a month to see how many 👍 votes the opening comment gets to help us make our decision.