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()
