X-Git-Url: https://plomlompom.com/repos/%22https:/validator.w3.org/static/git-favicon.png?a=blobdiff_plain;f=tests%2Futils.py;h=61dbb36b949ee3fdfd5a38dcada155a5ab18a924;hb=a1d3ef4b42bdac28f20d7f104ff93bf7efa37a30;hp=a826c16a5adb1d24222ef1f4b42f5cefc711baf8;hpb=f8148a40996d64878431ac59d47650741e1d79c5;p=plomtask diff --git a/tests/utils.py b/tests/utils.py index a826c16..61dbb36 100644 --- 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))