26
26
27
27
28
28
class CustomMock :
29
- def __init__ (self , json_info ):
29
+ def __init__ (self , json_info , id = "" ):
30
30
self .json_info = json_info
31
+ self .id = id
31
32
32
33
def get (self , key1 , key2 = None , key3 = None ):
33
34
if key2 is None and key3 is None :
@@ -69,7 +70,7 @@ def test_create_score_definition():
69
70
with pytest .raises (HTTPError ):
70
71
sd .create_score_definition (
71
72
score_def_name = "test_create_sd" ,
72
- model_id = "12345" ,
73
+ model = "12345" ,
73
74
table_name = "test_table" ,
74
75
)
75
76
@@ -87,7 +88,7 @@ def test_create_score_definition():
87
88
with pytest .raises (HTTPError ):
88
89
sd .create_score_definition (
89
90
score_def_name = "test_create_sd" ,
90
- model_id = "12345" ,
91
+ model = "12345" ,
91
92
table_name = "test_table" ,
92
93
)
93
94
@@ -98,7 +99,7 @@ def test_create_score_definition():
98
99
with pytest .raises (HTTPError ):
99
100
sd .create_score_definition (
100
101
score_def_name = "test_create_sd" ,
101
- model_id = "12345" ,
102
+ model = "12345" ,
102
103
table_name = "test_table" ,
103
104
table_file = "test_path" ,
104
105
)
@@ -112,7 +113,7 @@ def test_create_score_definition():
112
113
get_table .return_value = get_table_mock
113
114
response = sd .create_score_definition (
114
115
score_def_name = "test_create_sd" ,
115
- model_id = "12345" ,
116
+ model = "12345" ,
116
117
table_name = "test_table" ,
117
118
table_file = "test_path" ,
118
119
)
@@ -122,7 +123,7 @@ def test_create_score_definition():
122
123
get_table .return_value = get_table_mock
123
124
response = sd .create_score_definition (
124
125
score_def_name = "test_create_sd" ,
125
- model_id = "12345" ,
126
+ model = "12345" ,
126
127
table_name = "test_table" ,
127
128
table_file = "test_path" ,
128
129
)
@@ -146,7 +147,7 @@ def test_create_score_definition():
146
147
get_table .return_value = get_table_mock
147
148
response = sd .create_score_definition (
148
149
score_def_name = "test_create_sd" ,
149
- model_id = "12345" ,
150
+ model = "12345" ,
150
151
table_name = "test_table" ,
151
152
)
152
153
assert response
0 commit comments