176b903eeSLucas Bates""" 276b903eeSLucas Batestdc_helper.py - tdc helper functions 376b903eeSLucas Bates 476b903eeSLucas BatesCopyright (C) 2017 Lucas Bates <lucasb@mojatatu.com> 576b903eeSLucas Bates""" 676b903eeSLucas Bates 776b903eeSLucas Batesdef get_categorized_testlist(alltests, ucat): 876b903eeSLucas Bates """ Sort the master test list into categories. """ 976b903eeSLucas Bates testcases = dict() 1076b903eeSLucas Bates 1176b903eeSLucas Bates for category in ucat: 1276b903eeSLucas Bates testcases[category] = list(filter(lambda x: category in x['category'], alltests)) 1376b903eeSLucas Bates 1476b903eeSLucas Bates return(testcases) 1576b903eeSLucas Bates 1676b903eeSLucas Bates 1776b903eeSLucas Batesdef get_unique_item(lst): 187ae677fbSBrenda J. Butler """ For a list, return a list of the unique items in the list. """ 1976b903eeSLucas Bates return list(set(lst)) 2076b903eeSLucas Bates 2176b903eeSLucas Bates 2276b903eeSLucas Batesdef get_test_categories(alltests): 2376b903eeSLucas Bates """ Discover all unique test categories present in the test case file. """ 2476b903eeSLucas Bates ucat = [] 2576b903eeSLucas Bates for t in alltests: 2676b903eeSLucas Bates ucat.extend(get_unique_item(t['category'])) 2776b903eeSLucas Bates ucat = get_unique_item(ucat) 2876b903eeSLucas Bates return ucat 2976b903eeSLucas Bates 3076b903eeSLucas Batesdef list_test_cases(testlist): 3176b903eeSLucas Bates """ Print IDs and names of all test cases. """ 3276b903eeSLucas Bates for curcase in testlist: 3376b903eeSLucas Bates print(curcase['id'] + ': (' + ', '.join(curcase['category']) + ") " + curcase['name']) 3476b903eeSLucas Bates 3576b903eeSLucas Bates 3676b903eeSLucas Batesdef list_categories(testlist): 3776b903eeSLucas Bates """ Show all categories that are present in a test case file. """ 3876b903eeSLucas Bates categories = set(map(lambda x: x['category'], testlist)) 3976b903eeSLucas Bates print("Available categories:") 4076b903eeSLucas Bates print(", ".join(str(s) for s in categories)) 4176b903eeSLucas Bates print("") 4276b903eeSLucas Bates 4376b903eeSLucas Bates 4476b903eeSLucas Batesdef print_list(cmdlist): 4576b903eeSLucas Bates """ Print a list of strings prepended with a tab. """ 4676b903eeSLucas Bates for l in cmdlist: 4776b903eeSLucas Bates if (type(l) == list): 4876b903eeSLucas Bates print("\t" + str(l[0])) 4976b903eeSLucas Bates else: 5076b903eeSLucas Bates print("\t" + str(l)) 5176b903eeSLucas Bates 5276b903eeSLucas Bates 5376b903eeSLucas Batesdef print_sll(items): 5476b903eeSLucas Bates print("\n".join(str(s) for s in items)) 5576b903eeSLucas Bates 5676b903eeSLucas Bates 5776b903eeSLucas Batesdef print_test_case(tcase): 5876b903eeSLucas Bates """ Pretty-printing of a given test case. """ 5976b903eeSLucas Bates for k in tcase.keys(): 60*6c26c3fbSBrenda J. Butler if (isinstance(tcase[k], list)): 6176b903eeSLucas Bates print(k + ":") 6276b903eeSLucas Bates print_list(tcase[k]) 6376b903eeSLucas Bates else: 6476b903eeSLucas Bates print(k + ": " + tcase[k]) 6576b903eeSLucas Bates 6676b903eeSLucas Bates 6776b903eeSLucas Batesdef show_test_case_by_id(testlist, caseID): 6876b903eeSLucas Bates """ Find the specified test case to pretty-print. """ 6976b903eeSLucas Bates if not any(d.get('id', None) == caseID for d in testlist): 7076b903eeSLucas Bates print("That ID does not exist.") 7176b903eeSLucas Bates exit(1) 7276b903eeSLucas Bates else: 7376b903eeSLucas Bates print_test_case(next((d for d in testlist if d['id'] == caseID))) 7476b903eeSLucas Bates 7576b903eeSLucas Bates 76