"test/vscode:/vscode.git/clone" did not exist on "6b4c86ab8dc9820e8d307e6ccbefa94ea1e51844"
test_smartparam.py 3.13 KB
Newer Older
Deshui Yu's avatar
Deshui Yu committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# Copyright (c) Microsoft Corporation. All rights reserved.
#
# MIT License
#
# Permission is hereby granted, free of charge, to any person obtaining a copy of this software and
# associated documentation files (the "Software"), to deal in the Software without restriction,
# including without limitation the rights to use, copy, modify, merge, publish, distribute,
# sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all copies or
# substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT
# NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
# DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT
# OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
# ==================================================================================================

21
22
23
import os

os.environ['NNI_PLATFORM'] = 'unittest'
Deshui Yu's avatar
Deshui Yu committed
24
25
26
27
28
29
30
31
32
33
34
35

import nni
import nni.platform.test as test_platform
import nni.trial

from unittest import TestCase, main



class SmartParamTestCase(TestCase):
    def setUp(self):
        params = {
36
37
38
39
40
            'test_smartparam/choice1/choice': 'a',
            'test_smartparam/choice2/choice': '3*2+1',
            'test_smartparam/choice3/choice': '[1, 2]',
            'test_smartparam/choice4/choice': '{"a", 2}',
            'test_smartparam/func/function_choice': 'bar',
Zejun Lin's avatar
Zejun Lin committed
41
            'test_smartparam/lambda_func/function_choice': "lambda: 2*3"
Deshui Yu's avatar
Deshui Yu committed
42
43
44
45
46
        }
        nni.trial._params = { 'parameter_id': 'test_trial', 'parameters': params }


    def test_specified_name(self):
Zejun Lin's avatar
Zejun Lin committed
47
        val = nni.choice({'a': 'a', '3*2+1': 3*2+1, '[1, 2]': [1, 2], '{"a", 2}': {"a", 2}}, name = 'choice1', key='test_smartparam/choice1/choice')
48
        self.assertEqual(val, 'a')
Zejun Lin's avatar
Zejun Lin committed
49
        val = nni.choice({'a': 'a', '3*2+1': 3*2+1, '[1, 2]': [1, 2], '{"a", 2}': {"a", 2}}, name = 'choice2', key='test_smartparam/choice2/choice')
50
        self.assertEqual(val, 7)
Zejun Lin's avatar
Zejun Lin committed
51
        val = nni.choice({'a': 'a', '3*2+1': 3*2+1, '[1, 2]': [1, 2], '{"a", 2}': {"a", 2}}, name = 'choice3', key='test_smartparam/choice3/choice')
52
        self.assertEqual(val, [1, 2])
Zejun Lin's avatar
Zejun Lin committed
53
        val = nni.choice({'a': 'a', '3*2+1': 3*2+1, '[1, 2]': [1, 2], '{"a", 2}': {"a", 2}}, name = 'choice4', key='test_smartparam/choice4/choice')
54
        self.assertEqual(val, {"a", 2})
Deshui Yu's avatar
Deshui Yu committed
55

Zejun Lin's avatar
Zejun Lin committed
56
57
    def test_func(self):
        val = nni.function_choice({'foo': foo, 'bar': bar}, name='func', key='test_smartparam/func/function_choice')
Deshui Yu's avatar
Deshui Yu committed
58
59
        self.assertEqual(val, 'bar')

60
    def test_lambda_func(self):
Zejun Lin's avatar
Zejun Lin committed
61
        val = nni.function_choice({"lambda: 2*3": lambda: 2*3, "lambda: 3*4": lambda: 3*4}, name = 'lambda_func', key='test_smartparam/lambda_func/function_choice')
62
63
        self.assertEqual(val, 6)

Deshui Yu's avatar
Deshui Yu committed
64
65
66
67
68
69
70
71
72
73

def foo():
    return 'foo'

def bar():
    return 'bar'


if __name__ == '__main__':
    main()