IDSIA /
sacred
| @@ 177-190 (lines=14) @@ | ||
| 174 | assert tinydb_obs != 'foo' |
|
| 175 | ||
| 176 | ||
| 177 | def test_tinydb_observer_heartbeat_event_updates_run(tinydb_obs, sample_run): |
|
| 178 | tinydb_obs.started_event(**sample_run) |
|
| 179 | ||
| 180 | info = {'my_info': [1, 2, 3], 'nr': 7} |
|
| 181 | outp = 'some output' |
|
| 182 | tinydb_obs.heartbeat_event(info=info, captured_out=outp, beat_time=T2, |
|
| 183 | result=42) |
|
| 184 | ||
| 185 | assert len(tinydb_obs.runs) == 1 |
|
| 186 | db_run = tinydb_obs.runs.get(eid=1) |
|
| 187 | assert db_run['heartbeat'] == T2 |
|
| 188 | assert db_run['result'] == 42 |
|
| 189 | assert db_run['info'] == info |
|
| 190 | assert db_run['captured_out'] == outp |
|
| 191 | ||
| 192 | ||
| 193 | def test_tinydb_observer_completed_event_updates_run(tinydb_obs, sample_run): |
|
| @@ 90-103 (lines=14) @@ | ||
| 87 | assert mongo_obs != 'foo' |
|
| 88 | ||
| 89 | ||
| 90 | def test_mongo_observer_heartbeat_event_updates_run(mongo_obs, sample_run): |
|
| 91 | mongo_obs.started_event(**sample_run) |
|
| 92 | ||
| 93 | info = {'my_info': [1, 2, 3], 'nr': 7} |
|
| 94 | outp = 'some output' |
|
| 95 | mongo_obs.heartbeat_event(info=info, captured_out=outp, beat_time=T2, |
|
| 96 | result=1337) |
|
| 97 | ||
| 98 | assert mongo_obs.runs.count() == 1 |
|
| 99 | db_run = mongo_obs.runs.find_one() |
|
| 100 | assert db_run['heartbeat'] == T2 |
|
| 101 | assert db_run['result'] == 1337 |
|
| 102 | assert db_run['info'] == info |
|
| 103 | assert db_run['captured_out'] == outp |
|
| 104 | ||
| 105 | ||
| 106 | def test_mongo_observer_completed_event_updates_run(mongo_obs, sample_run): |
|
