def test_hard_delete_run(self):
fs = FileStore(self.test_root)
exp_id = self.experiments[random_int(0, len(self.experiments) - 1)]
run_id = self.exp_data[exp_id]["runs"][0]
fs._hard_delete_run(run_id)
with self.assertRaises(MlflowException):
fs.get_run(run_id)
with self.assertRaises(MlflowException):
fs.get_all_tags(run_id)
with self.assertRaises(MlflowException):
fs.get_all_metrics(run_id)
with self.assertRaises(MlflowException):
fs.get_all_params(run_id)
def test_get_all_metrics(self):
fs = FileStore(self.test_root)
for exp_id in self.experiments:
runs = self.exp_data[exp_id]["runs"]
for run_id in runs:
run_info = self.run_data[run_id]
metrics = fs.get_all_metrics(run_id)
metrics_dict = run_info.pop("metrics")
for metric in metrics:
expected_timestamp, expected_value = max(metrics_dict[metric.key])
self.assertEqual(metric.timestamp, expected_timestamp)
self.assertEqual(metric.value, expected_value)
