Allow passing arguments to unittest

test/framework/asyncprocess.py

@@ -67,9 +67,12 @@ def tearDown(self):
         super(AsyncProcessTest, self).tearDown()
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(AsyncProcessTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(AsyncProcessTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(AsyncProcessTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/build_log.py

@@ -437,9 +437,12 @@ def test_raise_nosupport(self):
                               raise_nosupport, 'foobar', 42)
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(BuildLogTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(BuildLogTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(BuildLogTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/config.py

@@ -731,8 +731,11 @@ def test_update_build_options(self):
         self.assertEqual(build_option('pr_target_account'), 'easybuilders')
 
 
-def suite():
-    return TestLoaderFiltered().loadTestsFromTestCase(EasyBuildConfigTest, sys.argv[1:])
+def suite(loader=None):
+    if loader:
+        return loader.loadTestsFromTestCase(EasyBuildConfigTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(EasyBuildConfigTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/containers.py

@@ -509,9 +509,12 @@ def test_container_config_template_recipe(self):
         self.assertEqual(cont_recipe, expected)
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(ContainersTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(ContainersTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(ContainersTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/docs.py

@@ -1067,9 +1067,12 @@ def get_eb_help_output(arg=''):
             self.assertTrue(regex.search(txt_rst), "Pattern '%s' should be found in: %s" % (regex.pattern, txt_rst))
 
 
-def suite():
+def suite(loader=None):
     """ returns all test cases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(DocsTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(DocsTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(DocsTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/easyblock.py

@@ -2839,9 +2839,12 @@ def run_sanity_check_step(sanity_check_paths, enhance_sanity_check):
         run_sanity_check_step({}, True)
 
 
-def suite():
+def suite(loader=None):
     """ return all the tests in this file """
-    return TestLoaderFiltered().loadTestsFromTestCase(EasyBlockTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(EasyBlockTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(EasyBlockTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/easyconfig.py

@@ -4883,9 +4883,12 @@ def test_easyconfigs_caches(self):
         self.assertTrue(regex.search(stdout), "Pattern '%s' should be found in: %s" % (regex.pattern, stdout))
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(EasyConfigTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(EasyConfigTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(EasyConfigTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/easyconfigformat.py

@@ -46,9 +46,12 @@ def test_parser_version_regex(self):
         self.assertEqual(version['minor'], int(res['minor']))
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(EasyConfigFormatTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(EasyConfigFormatTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(EasyConfigFormatTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/easyconfigparser.py

@@ -223,9 +223,12 @@ def test_check_value_types(self):
         self.assertEqual(ecdict['version'], '1.4')
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(EasyConfigParserTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(EasyConfigParserTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(EasyConfigParserTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/easyconfigversion.py

@@ -285,9 +285,12 @@ def test_hashing(self):
             self.assertTrue(hash(test_case))
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(EasyConfigVersion, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(EasyConfigVersion)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(EasyConfigVersion, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/easystack.py

@@ -175,9 +175,12 @@ def test_check_value(self):
             self.assertErrorRegex(EasyBuildError, error_pattern, check_value, version, context)
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(EasyStackTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(EasyStackTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(EasyStackTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/ebconfigobj.py

@@ -262,9 +262,12 @@ def test_ebconfigobj(self):
         self.assertEqual(res, {'foo': 'bar'})
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(TestEBConfigObj, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(TestEBConfigObj)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(TestEBConfigObj, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/environment.py

@@ -163,9 +163,12 @@ def test_sanitize_env(self):
         self.assertEqual(os.getenv('LD_PRELOAD'), None)
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(EnvironmentTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(EnvironmentTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(EnvironmentTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/filetools.py

@@ -3428,9 +3428,12 @@ def test_create_unused_dir(self):
         self.assertExists(path)
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(FileToolsTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(FileToolsTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(FileToolsTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/format_convert.py

@@ -60,9 +60,12 @@ def test_dependency(self):
         self.assertEqual(str(res), txt)
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(ConvertTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(ConvertTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(ConvertTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/general.py

@@ -155,9 +155,12 @@ def test_import_available_modules(self):
         self.assertEqual([test123.one, test123.three, test123.two], res)
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(GeneralTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(GeneralTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(GeneralTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/github.py

@@ -1249,9 +1249,12 @@ def test_is_patch_for(self):
         self.assertTrue(is_patch_for('pi.patch', ec))
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(GithubTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(GithubTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(GithubTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/hooks.py

@@ -217,9 +217,12 @@ def test_verify_hooks(self):
         self.assertErrorRegex(EasyBuildError, error_msg_pattern, load_hooks, test_broken_hooks_pymod)
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(HooksTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(HooksTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(HooksTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/include.py

@@ -258,9 +258,12 @@ def test_is_software_specific_easyblock(self):
         self.assertFalse(is_software_specific_easyblock(os.path.join(test_easyblocks, 'generic', 'toolchain.py')))
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(IncludeTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(IncludeTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(IncludeTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/lib.py

@@ -124,8 +124,11 @@ def test_modules_tool(self):
         self.assertEqual(modtool.list(), [{'default': None, 'mod_name': 'GCC/6.4.0-2.28'}])
 
 
-def suite():
-    return TestLoaderFiltered().loadTestsFromTestCase(EasyBuildLibTest, sys.argv[1:])
+def suite(loader=None):
+    if loader:
+        return loader.loadTestsFromTestCase(EasyBuildLibTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(EasyBuildLibTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/license.py

@@ -67,9 +67,12 @@ def test_licenses(self):
             self.assertTrue(issubclass(lics[lic], License))
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(LicenseTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(LicenseTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(LicenseTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/module_generator.py

@@ -1599,11 +1599,16 @@ class LuaModuleGeneratorTest(ModuleGeneratorTest):
     MODULE_GENERATOR_CLASS = ModuleGeneratorLua
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
+    if loader:
+        args = []
+    else:
+        loader = TestLoaderFiltered()
+        args = [sys.argv[1:]]
     suite = TestSuite()
-    suite.addTests(TestLoaderFiltered().loadTestsFromTestCase(TclModuleGeneratorTest, sys.argv[1:]))
-    suite.addTests(TestLoaderFiltered().loadTestsFromTestCase(LuaModuleGeneratorTest, sys.argv[1:]))
+    suite.addTests(loader.loadTestsFromTestCase(TclModuleGeneratorTest, *args))
+    suite.addTests(loader.loadTestsFromTestCase(LuaModuleGeneratorTest, *args))
     return suite
 
 

test/framework/modules.py

@@ -1546,9 +1546,12 @@ def test_get_setenv_value_from_modulefile(self):
         self.assertEqual(res, None)
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(ModulesTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(ModulesTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(ModulesTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/modulestool.py

@@ -204,9 +204,12 @@ def tearDown(self):
                 del os.environ['module']
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(ModulesToolTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(ModulesToolTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(ModulesToolTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/options.py

@@ -6870,9 +6870,12 @@ def test_opts_dict_to_eb_opts(self):
         self.assertEqual(opts_dict_to_eb_opts(opts_dict), expected)
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(CommandLineOptionsTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(CommandLineOptionsTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(CommandLineOptionsTest, sys.argv[1:])
 
 
 if __name__ == '__main__':

test/framework/output.py

@@ -181,9 +181,12 @@ def test_get_start_update_stop_progress_bar(self):
         stop_progress_bar(PROGRESS_BAR_EXTENSIONS)
 
 
-def suite():
+def suite(loader=None):
     """ returns all the testcases in this module """
-    return TestLoaderFiltered().loadTestsFromTestCase(OutputTest, sys.argv[1:])
+    if loader:
+        return loader.loadTestsFromTestCase(OutputTest)
+    else:
+        return TestLoaderFiltered().loadTestsFromTestCase(OutputTest, sys.argv[1:])
 
 
 if __name__ == '__main__':