misc.py 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. #
  2. # Copyright (C) 2016 Intel Corporation
  3. #
  4. # SPDX-License-Identifier: MIT
  5. #
  6. def toList(obj, obj_type, obj_name="Object"):
  7. if isinstance(obj, obj_type):
  8. return [obj]
  9. elif isinstance(obj, list):
  10. return obj
  11. else:
  12. raise TypeError("%s must be %s or list" % (obj_name, obj_type))
  13. def toSet(obj, obj_type, obj_name="Object"):
  14. if isinstance(obj, obj_type):
  15. return {obj}
  16. elif isinstance(obj, list):
  17. return set(obj)
  18. elif isinstance(obj, set):
  19. return obj
  20. else:
  21. raise TypeError("%s must be %s or set" % (obj_name, obj_type))
  22. def strToList(obj, obj_name="Object"):
  23. return toList(obj, str, obj_name)
  24. def strToSet(obj, obj_name="Object"):
  25. return toSet(obj, str, obj_name)
  26. def intToList(obj, obj_name="Object"):
  27. return toList(obj, int, obj_name)
  28. def dataStoteToDict(d, variables):
  29. data = {}
  30. for v in variables:
  31. data[v] = d.getVar(v)
  32. return data
  33. def updateTestData(d, td, variables):
  34. """
  35. Updates variables with values of data store to test data.
  36. """
  37. for var in variables:
  38. td[var] = d.getVar(var)