| """ |
| # SPDX-License-Identifier: GPL-2.0 |
| tdc_helper.py - tdc helper functions |
| |
| Copyright (C) 2017 Lucas Bates <lucasb@mojatatu.com> |
| """ |
| |
| def get_categorized_testlist(alltests, ucat): |
| """ Sort the master test list into categories. """ |
| testcases = dict() |
| |
| for category in ucat: |
| testcases[category] = list(filter(lambda x: category in x['category'], alltests)) |
| |
| return(testcases) |
| |
| |
| def get_unique_item(lst): |
| """ For a list, return a list of the unique items in the list. """ |
| if len(lst) > 1: |
| return list(set(lst)) |
| else: |
| return lst |
| |
| |
| def get_test_categories(alltests): |
| """ Discover all unique test categories present in the test case file. """ |
| ucat = [] |
| for t in alltests: |
| ucat.extend(get_unique_item(t['category'])) |
| ucat = get_unique_item(ucat) |
| return ucat |
| |
| def list_test_cases(testlist): |
| """ Print IDs and names of all test cases. """ |
| for curcase in testlist: |
| print(curcase['id'] + ': (' + ', '.join(curcase['category']) + ") " + curcase['name']) |
| |
| |
| def list_categories(testlist): |
| """ Show all categories that are present in a test case file. """ |
| categories = set(map(lambda x: x['category'], testlist)) |
| print("Available categories:") |
| print(", ".join(str(s) for s in categories)) |
| print("") |
| |
| |
| def print_list(cmdlist): |
| """ Print a list of strings prepended with a tab. """ |
| for l in cmdlist: |
| if (type(l) == list): |
| print("\t" + str(l[0])) |
| else: |
| print("\t" + str(l)) |
| |
| |
| def print_sll(items): |
| print("\n".join(str(s) for s in items)) |
| |
| |
| def print_test_case(tcase): |
| """ Pretty-printing of a given test case. """ |
| print('\n==============\nTest {}\t{}\n'.format(tcase['id'], tcase['name'])) |
| for k in tcase.keys(): |
| if (isinstance(tcase[k], list)): |
| print(k + ":") |
| print_list(tcase[k]) |
| else: |
| if not ((k == 'id') or (k == 'name')): |
| print(k + ": " + str(tcase[k])) |
