def test_delete_figure(self): """Test deleting a figure.""" exp_data = DbExperimentData(experiment_type="qiskit_test") id_template = "figure_{}.svg" for idx in range(3): exp_data.add_figures(str.encode("hello world"), id_template.format(idx)) sub_tests = [(1, id_template.format(1)), (id_template.format(2), id_template.format(2))] for del_key, figure_name in sub_tests: with self.subTest(del_key=del_key): exp_data.delete_figure(del_key) self.assertRaises(DbExperimentEntryNotFound, exp_data.figure, figure_name)
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()