71 lines · python
1"""2# SPDX-License-Identifier: GPL-2.03tdc_helper.py - tdc helper functions4 5Copyright (C) 2017 Lucas Bates <lucasb@mojatatu.com>6"""7 8def get_categorized_testlist(alltests, ucat):9 """ Sort the master test list into categories. """10 testcases = dict()11 12 for category in ucat:13 testcases[category] = list(filter(lambda x: category in x['category'], alltests))14 15 return(testcases)16 17 18def get_unique_item(lst):19 """ For a list, return a list of the unique items in the list. """20 if len(lst) > 1:21 return list(set(lst))22 else:23 return lst24 25 26def get_test_categories(alltests):27 """ Discover all unique test categories present in the test case file. """28 ucat = []29 for t in alltests:30 ucat.extend(get_unique_item(t['category']))31 ucat = get_unique_item(ucat)32 return ucat33 34def list_test_cases(testlist):35 """ Print IDs and names of all test cases. """36 for curcase in testlist:37 print(curcase['id'] + ': (' + ', '.join(curcase['category']) + ") " + curcase['name'])38 39 40def list_categories(testlist):41 """ Show all categories that are present in a test case file. """42 categories = set(map(lambda x: x['category'], testlist))43 print("Available categories:")44 print(", ".join(str(s) for s in categories))45 print("")46 47 48def print_list(cmdlist):49 """ Print a list of strings prepended with a tab. """50 for l in cmdlist:51 if (type(l) == list):52 print("\t" + str(l[0]))53 else:54 print("\t" + str(l))55 56 57def print_sll(items):58 print("\n".join(str(s) for s in items))59 60 61def print_test_case(tcase):62 """ Pretty-printing of a given test case. """63 print('\n==============\nTest {}\t{}\n'.format(tcase['id'], tcase['name']))64 for k in tcase.keys():65 if (isinstance(tcase[k], list)):66 print(k + ":")67 print_list(tcase[k])68 else:69 if not ((k == 'id') or (k == 'name')):70 print(k + ": " + str(tcase[k]))71