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)