Commit 9e40371c authored by suiguoxin's avatar suiguoxin
Browse files

add doc and test

parent a4d62e95
......@@ -8,6 +8,8 @@ Trial
.. autofunction:: nni.get_current_parameter
.. autofunction:: nni.report_intermediate_result
.. autofunction:: nni.report_final_result
.. autofunction:: nni.get_experiment_id
.. autofunction:: nni.get_trial_id
.. autofunction:: nni.get_sequence_id
......
......@@ -32,10 +32,13 @@ _last_metric = None
def get_next_parameter():
return _params
def get_sequence_id():
return 0
def get_experiment_id():
return 'fakeidex'
def get_trial_id():
return 'fakeidtr'
def get_sequence_id():
return 0
def send_metric(string):
......
......@@ -38,6 +38,12 @@ class TrialTestCase(TestCase):
nni.get_next_parameter()
self.assertEqual(nni.get_current_parameter('x'), 123)
def test_get_experiment_id(self):
self.assertEqual(nni.get_experiment_id(), 'fakeidex')
def test_get_trial_id(self):
self.assertEqual(nni.get_trial_id(), 'fakeidtr')
def test_get_sequence_id(self):
self.assertEqual(nni.get_sequence_id(), 0)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment