Minor tests refactoring.

[plomtask] / tests / utils.py

diff --git a/tests/utils.py b/tests/utils.py
index a826c16a5adb1d24222ef1f4b42f5cefc711baf8..61dbb36b949ee3fdfd5a38dcada155a5ab18a924 100644 (file)
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -79,8 +79,9 @@ class TestCaseWithDB(TestCase):
         self.assertEqual(self.checked_class(id2), by_id_created)
         self.check_storage([obj])
 
-    def check_from_table_row(self, id_: int | str) -> None:
+    def check_from_table_row(self) -> None:
         """Test .from_table_row() properly reads in class from DB"""
+        id_ = self.default_ids[0]
         obj = self.checked_class(id_)  # pylint: disable=not-callable
         obj.save(self.db_conn)
         assert isinstance(obj.id_, (str, int))