#
|
# Copyright (C) 2016 Intel Corporation
|
#
|
# SPDX-License-Identifier: MIT
|
#
|
|
def toList(obj, obj_type, obj_name="Object"):
|
if isinstance(obj, obj_type):
|
return [obj]
|
elif isinstance(obj, list):
|
return obj
|
else:
|
raise TypeError("%s must be %s or list" % (obj_name, obj_type))
|
|
def toSet(obj, obj_type, obj_name="Object"):
|
if isinstance(obj, obj_type):
|
return {obj}
|
elif isinstance(obj, list):
|
return set(obj)
|
elif isinstance(obj, set):
|
return obj
|
else:
|
raise TypeError("%s must be %s or set" % (obj_name, obj_type))
|
|
def strToList(obj, obj_name="Object"):
|
return toList(obj, str, obj_name)
|
|
def strToSet(obj, obj_name="Object"):
|
return toSet(obj, str, obj_name)
|
|
def intToList(obj, obj_name="Object"):
|
return toList(obj, int, obj_name)
|
|
def dataStoteToDict(d, variables):
|
data = {}
|
|
for v in variables:
|
data[v] = d.getVar(v)
|
|
return data
|
|
def updateTestData(d, td, variables):
|
"""
|
Updates variables with values of data store to test data.
|
"""
|
for var in variables:
|
td[var] = d.getVar(var)
|