Don't need __model__ declaration in builder classes.

Author: acerathereinspirative-nhs-nvir

tests/utils/builders.py

@@ -7,22 +7,18 @@
 from eligibility_signposting_api.model.rules import CampaignConfig, Iteration, IterationCohort, IterationRule
 
 
-class IterationCohortFactory(ModelFactory[IterationCohort]):
-    __model__ = IterationCohort
+class IterationCohortFactory(ModelFactory[IterationCohort]): ...
 
 
-class IterationRuleFactory(ModelFactory[IterationRule]):
-    __model__ = IterationRule
+class IterationRuleFactory(ModelFactory[IterationRule]): ...
 
 
 class IterationFactory(ModelFactory[Iteration]):
-    __model__ = Iteration
     iteration_cohorts = Use(IterationCohortFactory.batch, size=2)
     iteration_rules = Use(IterationRuleFactory.batch, size=2)
 
 
 class CampaignConfigFactory(ModelFactory[CampaignConfig]):
-    __model__ = CampaignConfig
     iterations = Use(IterationFactory.batch, size=2)
 
 

tests/utils/matchers/meta.py

@@ -60,11 +60,8 @@ def is_eligibility_status() -> Matcher[EligibilityStatus]: return EligibilitySta
 
     __domain_class__ = None  # Will be inferred when subclassed generically
 
-    def __init__(self):
-        super().__init__()
-
     def describe_to(self, description: Description) -> None:
-        description.append_text(f"{self.__class__.__name__.removesuffix('Matcher')} with")
+        description.append_text(f"{self.__domain_class__.__name__} with")
         for field_name in self.__domain_class__.__annotations__:
             attr_name = f"{field_name}_" if field_name in {"id", "type"} else field_name
             self.append_matcher_description(getattr(self, attr_name), field_name, description)