fix chooser script model checker (#515)

Author: ben-c-at-moz

choose_ci_set.py

@@ -122,7 +122,7 @@ def dedupe(run_list: list, slash: str) -> list:
         "suite_conftest_re_string": r".*/.*/conftest\.py",
         "selectors_json_re_string": r"modules/data/.*\.components\.json",
         "object_model_re_string": r"modules/.*.object.*\.py",
-        "class_re_string": r"\s*class (\w+):",
+        "class_re_string": r"\s*class (\w+)[(A-Za-z0-9_)]*:",
     }
     for k in list(re_obj.keys()):
         if slash == "\\":

modules/page_object_generics.py

@@ -74,6 +74,7 @@ def select_toolbar_option(self, option: str) -> BasePage:
         return self
 
     def add_image(self, image_path: str, sys_platform: str) -> BasePage:
+        """Add an image to a pdf file"""
         self.get_element("toolbar-add-image").click()
         self.get_element("toolbar-add-image-confirm").click()
         sleep(3)