def testGettingClassesByAnnotations(self): module = sys.modules[__name__] ats = Annotations.getAnnotatedClasses(module, Test) self.assertEqual(len(ats), 2) data = set((a[0].data for c, a in ats)) self.assertSetEqual(data, {None, "bar"})
def scan(self, module): contracts = [c for c, a in Annotations.getAnnotatedClasses(module, Contract)] services = [c for c, a in Annotations.getAnnotatedClasses(module, Service)] return contracts, services