File tree 3 files changed +6
-6
lines changed
3 files changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -13,14 +13,14 @@ class ExperimentModel(BaseModel):
13
13
14
14
15
15
class ExperimentDBModel (ExperimentModel ):
16
- id : int
16
+ id : str
17
17
is_running : bool
18
18
started : Optional [int ] = None
19
19
20
20
21
21
class ExperimentDetailsDBModel (ExperimentDBModel ):
22
- snapshots : List [tuple ]
23
- temperatures : List [tuple ]
22
+ snapshots : List [tuple ] = []
23
+ temperatures : List [tuple ] = []
24
24
25
25
26
26
class DatabaseIface (ABC ):
Original file line number Diff line number Diff line change @@ -15,7 +15,7 @@ def __init__(self):
15
15
def add (self , experiment : ExperimentModel ) -> int :
16
16
experiment_id = len (self .experiments ) + 1
17
17
new_item = ExperimentDBModel (
18
- id = experiment_id ,
18
+ id = str ( experiment_id ) ,
19
19
is_running = True ,
20
20
started = int (datetime .now ().timestamp ()),
21
21
** experiment .dict ()
Original file line number Diff line number Diff line change @@ -60,7 +60,7 @@ def test_get_existing_experiment(client):
60
60
response = client .get ("/experiments/1" )
61
61
assert response .status_code == 200
62
62
assert response .json () == {
63
- 'id' : 1 ,
63
+ 'id' : '1' ,
64
64
'is_running' : True ,
65
65
** experiment_data ,
66
66
'started' : fake_datetime ,
@@ -81,7 +81,7 @@ def test_list_past_experiments(client):
81
81
response = client .get ("/experiments" )
82
82
assert response .status_code == 200
83
83
assert response .json () == [{
84
- 'id' : 1 ,
84
+ 'id' : '1' ,
85
85
'is_running' : True ,
86
86
'started' : fake_datetime ,
87
87
** experiment_data
You can’t perform that action at this time.
0 commit comments