Skip to content

Commit 869e83a

Browse files
test(nlu): update unit tests for hand edits
1 parent c24aac5 commit 869e83a

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

test/unit/test_natural_language_understanding_v1.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -577,11 +577,13 @@ def test_create_categories_model_required_params(self):
577577
# Set up parameter values
578578
language = 'testString'
579579
training_data = io.BytesIO(b'This is a mock file.').getvalue()
580+
training_data_content_type = 'application/json'
580581

581582
# Invoke method
582583
response = _service.create_categories_model(
583584
language,
584585
training_data,
586+
training_data_content_type,
585587
headers={}
586588
)
587589

@@ -615,11 +617,13 @@ def test_create_categories_model_value_error(self):
615617
# Set up parameter values
616618
language = 'testString'
617619
training_data = io.BytesIO(b'This is a mock file.').getvalue()
620+
training_data_content_type = 'application/json'
618621

619622
# Pass in all but one required param and check for a ValueError
620623
req_param_dict = {
621624
"language": language,
622625
"training_data": training_data,
626+
"training_data_content_type": training_data_content_type,
623627
}
624628
for param in req_param_dict.keys():
625629
req_copy = {key:val if key is not param else None for (key,val) in req_param_dict.items()}
@@ -853,12 +857,14 @@ def test_update_categories_model_required_params(self):
853857
model_id = 'testString'
854858
language = 'testString'
855859
training_data = io.BytesIO(b'This is a mock file.').getvalue()
860+
training_data_content_type = 'application/json'
856861

857862
# Invoke method
858863
response = _service.update_categories_model(
859864
model_id,
860865
language,
861866
training_data,
867+
training_data_content_type,
862868
headers={}
863869
)
864870

@@ -893,12 +899,14 @@ def test_update_categories_model_value_error(self):
893899
model_id = 'testString'
894900
language = 'testString'
895901
training_data = io.BytesIO(b'This is a mock file.').getvalue()
902+
training_data_content_type = 'application/json'
896903

897904
# Pass in all but one required param and check for a ValueError
898905
req_param_dict = {
899906
"model_id": model_id,
900907
"language": language,
901908
"training_data": training_data,
909+
"training_data_content_type": training_data_content_type,
902910
}
903911
for param in req_param_dict.keys():
904912
req_copy = {key:val if key is not param else None for (key,val) in req_param_dict.items()}
@@ -1078,11 +1086,13 @@ def test_create_classifications_model_required_params(self):
10781086
# Set up parameter values
10791087
language = 'testString'
10801088
training_data = io.BytesIO(b'This is a mock file.').getvalue()
1089+
training_data_content_type = 'application/json'
10811090

10821091
# Invoke method
10831092
response = _service.create_classifications_model(
10841093
language,
10851094
training_data,
1095+
training_data_content_type,
10861096
headers={}
10871097
)
10881098

@@ -1116,11 +1126,13 @@ def test_create_classifications_model_value_error(self):
11161126
# Set up parameter values
11171127
language = 'testString'
11181128
training_data = io.BytesIO(b'This is a mock file.').getvalue()
1129+
training_data_content_type = 'application/json'
11191130

11201131
# Pass in all but one required param and check for a ValueError
11211132
req_param_dict = {
11221133
"language": language,
11231134
"training_data": training_data,
1135+
"training_data_content_type": training_data_content_type,
11241136
}
11251137
for param in req_param_dict.keys():
11261138
req_copy = {key:val if key is not param else None for (key,val) in req_param_dict.items()}
@@ -1360,12 +1372,14 @@ def test_update_classifications_model_required_params(self):
13601372
model_id = 'testString'
13611373
language = 'testString'
13621374
training_data = io.BytesIO(b'This is a mock file.').getvalue()
1375+
training_data_content_type = 'application/json'
13631376

13641377
# Invoke method
13651378
response = _service.update_classifications_model(
13661379
model_id,
13671380
language,
13681381
training_data,
1382+
training_data_content_type,
13691383
headers={}
13701384
)
13711385

@@ -1400,12 +1414,14 @@ def test_update_classifications_model_value_error(self):
14001414
model_id = 'testString'
14011415
language = 'testString'
14021416
training_data = io.BytesIO(b'This is a mock file.').getvalue()
1417+
training_data_content_type = 'application/json'
14031418

14041419
# Pass in all but one required param and check for a ValueError
14051420
req_param_dict = {
14061421
"model_id": model_id,
14071422
"language": language,
14081423
"training_data": training_data,
1424+
"training_data_content_type": training_data_content_type
14091425
}
14101426
for param in req_param_dict.keys():
14111427
req_copy = {key:val if key is not param else None for (key,val) in req_param_dict.items()}

0 commit comments

Comments
 (0)