diff --git a/src/classes/play_classes.py b/src/classes/play_classes.py index 1ecf3a5..395483c 100644 --- a/src/classes/play_classes.py +++ b/src/classes/play_classes.py @@ -22,4 +22,8 @@ class Play(SQLModel, table=True): ignore_for_stats : bool location: str - players: list[PlayPlayer] = Relationship(back_populates="play") \ No newline at end of file + players: list[PlayPlayer] = Relationship(back_populates="play") + + model_config = { + 'validate_assignment':True + } diff --git a/tests/test_main.py b/tests/test_main.py index 5cd8d41..a28c9f4 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -68,8 +68,6 @@ def test_retrieve_plays(): returned_play = play_classes.Play(**response.json()[0]) assert type(returned_play.boardgame_id) == int - assert type(returned_play.players) == list - assert type(returned_play.players[0]) == play_classes.PlayPlayer assert type(returned_play.play_date) == date assert type(returned_play.duration) == int assert type(returned_play.ignore_for_stats) == bool