def test_reset(self): registry = Registry() expected = DEFAULT_GENERATOR_NAMES registry.register('abstract', AbstractKeyGenerator) now_expected = DEFAULT_GENERATOR_NAMES + ['abstract'] self.assertEqual(set(registry.available()), set(now_expected)) registry.reset() self.assertEqual(set(registry.available()), set(expected))
def setUp(self): registry = Registry() registry.reset()