def test_save_delete(self): """Test saving all deletion.""" exp_data = DbExperimentData(backend=self.backend, experiment_type="qiskit_test") service = mock.create_autospec(DatabaseServiceV1, instance=True) exp_data.add_figures(str.encode("hello world")) exp_data.add_analysis_results(mock.MagicMock()) exp_data.delete_analysis_result(0) exp_data.delete_figure(0) exp_data.service = service exp_data.save() service.create_experiment.assert_called_once() service.delete_figure.assert_called_once() service.delete_analysis_result.assert_called_once()
def test_delete_analysis_result(self): """Test deleting analysis result.""" exp_data = DbExperimentData(experiment_type="qiskit_test") id_template = "result_{}" for idx in range(3): res = mock.MagicMock() res.result_id = id_template.format(idx) exp_data.add_analysis_results(res) subtests = [(0, id_template.format(0)), (id_template.format(2), id_template.format(2))] for del_key, res_id in subtests: with self.subTest(del_key=del_key): exp_data.delete_analysis_result(del_key) self.assertRaises(DbExperimentEntryNotFound, exp_data.analysis_results, res_id)