Design decision: move .get('input') to our class methods?

Author: ElSnoMan

tests/test_todo.py

@@ -12,10 +12,10 @@ def goto(self) -> 'TodoPage':
         return self
 
     def get_todo_by_name(self, name: str) -> Element:
-        return self.py.getx(f"//*[text()='{name}']").parent()
+        return self.py.getx(f"//*[text()='{name}']").parent().get('input')
 
     def get_all_todos(self) -> Elements:
-        return self.py.find("li[ng-repeat*='todo']")
+        return self.py.find("li[ng-repeat*='todo'] > input")
 
 
 @pytest.fixture
@@ -24,21 +24,21 @@ def page(py: Pylenium):
 
 
 def test_check_first_item(page: TodoPage):
-    checkbox = page.get_todo_by_name('First Item').get('input')
+    checkbox = page.get_todo_by_name('First Item')
     checkbox.click()
     assert checkbox.should().be_checked()
 
 
 def test_check_many_items(py: Pylenium, page: TodoPage):
     todos = page.get_all_todos()
     todo2, todo4 = todos[1], todos[3]
-    todo2.get('input').click()
-    todo4.get('input').click()
+    todo2.click()
+    todo4.click()
     assert py.contains('3 of 5 remaining')
 
 
 def test_check_all_items(py: Pylenium, page: TodoPage):
     for todo in page.get_all_todos():
-        todo.get('input').click()
+        todo.click()
 
     assert py.contains('0 of 5 remaining')