2 # SPDX-License-Identifier: GPL-2.0
3 tdc_helper.py - tdc helper functions
5 Copyright (C) 2017 Lucas Bates <lucasb@mojatatu.com>
8 def get_categorized_testlist(alltests, ucat):
9 """ Sort the master test list into categories. """
13 testcases[category] = list(filter(lambda x: category in x['category'], alltests))
18 def get_unique_item(lst):
19 """ For a list, return a set of the unique items in the list. """
23 def get_test_categories(alltests):
24 """ Discover all unique test categories present in the test case file. """
27 ucat.extend(get_unique_item(t['category']))
28 ucat = get_unique_item(ucat)
31 def list_test_cases(testlist):
32 """ Print IDs and names of all test cases. """
33 for curcase in testlist:
34 print(curcase['id'] + ': (' + ', '.join(curcase['category']) + ") " + curcase['name'])
37 def list_categories(testlist):
38 """ Show all categories that are present in a test case file. """
39 categories = set(map(lambda x: x['category'], testlist))
40 print("Available categories:")
41 print(", ".join(str(s) for s in categories))
45 def print_list(cmdlist):
46 """ Print a list of strings prepended with a tab. """
49 print("\t" + str(l[0]))
55 print("\n".join(str(s) for s in items))
58 def print_test_case(tcase):
59 """ Pretty-printing of a given test case. """
60 for k in tcase.keys():
61 if (type(tcase[k]) == list):
65 print(k + ": " + tcase[k])
68 def show_test_case_by_id(testlist, caseID):
69 """ Find the specified test case to pretty-print. """
70 if not any(d.get('id', None) == caseID for d in testlist):
71 print("That ID does not exist.")
74 print_test_case(next((d for d in testlist if d['id'] == caseID)))