Skip to content

Commit 18e0d61

Browse files
committed
Simplified functions correspondiong to one of the scenarios
1 parent f9e8387 commit 18e0d61

File tree

2 files changed

+5
-16
lines changed

2 files changed

+5
-16
lines changed

test/tck/steps/bolt_type_steps.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -251,7 +251,7 @@ def step_impl(context):
251251
@then('result should be node "(?P<n1>.+)" node "(?P<n2>.+)" and int "(?P<len>\d+)"')
252252
def step_impl(context, n1, n2, len):
253253
result = context.results["as_string"]
254-
given = tck_util.single_result_to_values(result)
254+
given = tck_util.result_to_set(result)
255255
expected = tck_util.node_node_int_to_values(n1, n2, len)
256256
if given != expected:
257257
raise Exception("Mixed response does not match given: %s expected: %s" % (given, expected))

test/tck/tck_util.py

Lines changed: 4 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -289,22 +289,11 @@ def text_string_to_set(table):
289289
return result_to_set(table)
290290

291291

292-
def single_result_to_values(result):
293-
l = []
294-
295-
for val in result[0].values():
296-
if not isinstance(val,list):
297-
val = [val]
298-
val = [Value(v) for v in val]
299-
l.append(tuple(val))
300-
return tuple(l)
301-
302-
303292
def node_node_int_to_values(n1, n2, len):
304-
n1 = tuple([Value(get_node(n1)[0])])
305-
n2 = tuple([Value(get_node(n2)[0])])
306-
len = tuple([Value(int(len))])
307-
return tuple([n1,n2,len])
293+
n1 = Value(get_node(n1)[0])
294+
n2 = Value(get_node(n2)[0])
295+
len = Value(int(len))
296+
return {n1, n2, len}
308297

309298

310299
class Value:

0 commit comments

Comments
 (0)