def test_load_classes_from_package(self): classes = dict(load_classes(TEST_DATA)) self.assertEqual(classes['PowFixture']['methods'], ['pow']) self.assertEqual(sorted(classes['DivFixture']['methods']), ['div', 'random_number'])
def test_load_classes_from_module(self): pow_fixture_module = os.path.join(TEST_DATA, 'pow_fixture.py') classes = dict(load_classes(pow_fixture_module)) self.assertEqual(list(classes.keys()), ['PowFixture']) self.assertEqual(classes['PowFixture']['methods'], ['pow'])