public
/
ottertune
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

modify tests

This commit is contained in:
yangdsh 2020-01-12 14:40:51 +00:00 committed by Dana Van Aken
parent d3c7bb661d
commit a3774af2c1
2 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
server/website/tests/test_views.py
View File

@ -149,7 +149,8 @@ class SessionViewsTests(TestCase):
'memory': '16',
'storage': '32',
'storage_type': 5,
'dbms': 1
'dbms': 1,
'hyper_parameters': '{}'
}
def setUp(self):

15
server/website/website/views.py
View File

@ -1457,29 +1457,36 @@ def create_test_website(request): # pylint: disable=unused-argument
password='ottertune_test_user')
test_project = Project.objects.create(user=test_user, name='ottertune_test_project',
creation_time=now(), last_update=now())
hyper_parameters = JSONUtil.dumps(utils.get_constants())
# create no tuning session
s1 = Session.objects.create(name='test_session_no_tuning', tuning_session='no_tuning_session',
dbms_id=1, hardware=test_hardware, project=test_project,
creation_time=now(), last_update=now(), user=test_user,
upload_code='ottertuneTestNoTuning')
upload_code='ottertuneTestNoTuning',
hyper_parameters=hyper_parameters)
set_default_knobs(s1)
# create gpr session
s2 = Session.objects.create(name='test_session_gpr', tuning_session='tuning_session',
dbms_id=1, hardware=test_hardware, project=test_project,
creation_time=now(), last_update=now(), algorithm=AlgorithmType.GPR,
upload_code='ottertuneTestTuningGPR', user=test_user)
upload_code='ottertuneTestTuningGPR', user=test_user,
hyper_parameters=hyper_parameters)
set_default_knobs(s2)
# create dnn session
s3 = Session.objects.create(name='test_session_dnn', tuning_session='tuning_session',
dbms_id=1, hardware=test_hardware, project=test_project,
creation_time=now(), last_update=now(), algorithm=AlgorithmType.DNN,
upload_code='ottertuneTestTuningDNN', user=test_user)
upload_code='ottertuneTestTuningDNN', user=test_user,
hyper_parameters=hyper_parameters)
set_default_knobs(s3)
# create ddpg session
s4 = Session.objects.create(name='test_session_ddpg', tuning_session='tuning_session',
dbms_id=1, hardware=test_hardware, project=test_project,
creation_time=now(), last_update=now(), user=test_user,
upload_code='ottertuneTestTuningDDPG', algorithm=AlgorithmType.DDPG)
upload_code='ottertuneTestTuningDDPG',
algorithm=AlgorithmType.DDPG,
hyper_parameters=hyper_parameters)
set_default_knobs(s4)
response = HttpResponse("Success: create test website successfully")
return response