def test_experiment_tracker_in_parts(test_engine, project_path): experiment = SingleThreadedExperiment( config=sample_config(), db_engine=test_engine, project_path=project_path, ) experiment.generate_matrices() experiment.train_and_test_models() with scoped_session(test_engine) as session: experiment_run = session.query(ExperimentRun).get(experiment.run_id) assert experiment_run.start_method == "generate_matrices"
def test_experiment_tracker_in_parts(test_engine, project_path): with mock.patch("triage.util.conf.open", side_effect=open_side_effect) as mock_file: experiment = SingleThreadedExperiment( config=sample_config(), db_engine=test_engine, project_path=project_path, ) experiment.generate_matrices() experiment.train_and_test_models() with scoped_session(test_engine) as session: experiment_run = session.query(TriageRun).get(experiment.run_id) assert experiment_run.start_method == "generate_matrices"