def test_whole_module_import(self): classes = TestLoader.get_class_names_from_module('tests.dummy_modules') expected = set([ "tests.dummy_modules.mod1.ModuleOneHalf", "tests.dummy_modules.mod1.ModuleTwoHalves", "tests.dummy_modules.bars.testset_subclass.TestSetSubclass", "tests.dummy_modules.bars.testset_subclass_subclass.TestSetSubclassSubclass", ]) self.assertEqual(classes, expected)