def _test_report_final_result(self, in_, out):
nni.report_final_result(in_)
self.assertEqual(test_platform.get_last_metric(), {
'parameter_id': 'test_param',
'trial_job_id': 'test_trial_job_id',
'type': 'FINAL',
'sequence': 0,
'value': out
})
def test_report_intermediate_result(self):
nni.report_intermediate_result(123)
self.assertEqual(test_platform.get_last_metric(), {
'parameter_id': 'test_param',
'trial_job_id': 'test_trial_job_id',
'type': 'PERIODICAL',
'sequence': 0,
'value': 123
})
def test_report_final_result_nparray(self):
arr = np.array([[1, 2, 3], [4, 5, 6]])
nni.report_final_result(arr)
out = test_platform.get_last_metric()
self.assertEqual(len(arr), 2)
self.assertEqual(len(arr[0]), 3)
self.assertEqual(len(arr[1]), 3)
self.assertEqual(arr[0][0], 1)
self.assertEqual(arr[0][1], 2)
self.assertEqual(arr[0][2], 3)
self.assertEqual(arr[1][0], 4)
self.assertEqual(arr[1][1], 5)
self.assertEqual(arr[1][2], 6)