diff --git a/tests/test_session.py b/tests/test_session.py index 6a85f199453fd230917b6e43b6631b53a7d015df..2e5eb44fbb7272b7c4a4ef7d32129320dc710256 100644 --- a/tests/test_session.py +++ b/tests/test_session.py @@ -25,7 +25,6 @@ class TestGetSessionObject(unittest.TestCase): def test_session_logfile_is_valid(self): self.assertTrue(exists(self.sesh.logfile), 'Session did not create a log file in the correct place') - self.assertTrue(exists(self.sesh.manifest_json), 'Session did not create a manifest JSON file in the correct place') def test_changing_session_root_creates_new_directory(self): from model_server.conf.defaults import root @@ -74,19 +73,6 @@ class TestGetSessionObject(unittest.TestCase): self.assertEqual(logs[1]['level'], 'WARNING') self.assertEqual(logs[-1]['message'], 'Initialized session') - def test_session_records_workflow(self): - di = WorkflowRunRecord( - model_id='test_model', - input_filepath='/test/input/directory', - output_filepath='/test/output/fi.le', - success=True, - timer_results={'start': 0.123}, - ) - self.sesh.record_workflow_run(di) - with open(self.sesh.manifest_json, 'r') as fh: - do = json.load(fh) - self.assertEqual(di.dict(), do, 'Manifest record is not correct') - def test_session_loads_model(self): MC = DummySemanticSegmentationModel success = self.sesh.load_model(MC)