From 01d8c769f8acb02f09b572352b4ec0dfbf70a150 Mon Sep 17 00:00:00 2001 From: Yarne Coppens Date: Sun, 11 Aug 2024 21:45:27 +0200 Subject: [PATCH] Play class date is now a date type --- src/classes/play_classes.py | 6 +++++- tests/test_main.py | 2 -- 2 files changed, 5 insertions(+), 3 deletions(-) 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