1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889 |
- #
- # Copyright (C) 2016 Intel Corporation
- #
- # SPDX-License-Identifier: MIT
- #
- import os
- import inspect
- import unittest
- def getSuiteCases(suite):
- """
- Returns individual test from a test suite.
- """
- tests = []
- if isinstance(suite, unittest.TestCase):
- tests.append(suite)
- elif isinstance(suite, unittest.suite.TestSuite):
- for item in suite:
- tests.extend(getSuiteCases(item))
- return tests
- def getSuiteModules(suite):
- """
- Returns modules in a test suite.
- """
- modules = set()
- for test in getSuiteCases(suite):
- modules.add(getCaseModule(test))
- return modules
- def getSuiteCasesInfo(suite, func):
- """
- Returns test case info from suite. Info is fetched from func.
- """
- tests = []
- for test in getSuiteCases(suite):
- tests.append(func(test))
- return tests
- def getSuiteCasesNames(suite):
- """
- Returns test case names from suite.
- """
- return getSuiteCasesInfo(suite, getCaseMethod)
- def getSuiteCasesIDs(suite):
- """
- Returns test case ids from suite.
- """
- return getSuiteCasesInfo(suite, getCaseID)
- def getSuiteCasesFiles(suite):
- """
- Returns test case files paths from suite.
- """
- return getSuiteCasesInfo(suite, getCaseFile)
- def getCaseModule(test_case):
- """
- Returns test case module name.
- """
- return test_case.__module__
- def getCaseClass(test_case):
- """
- Returns test case class name.
- """
- return test_case.__class__.__name__
- def getCaseID(test_case):
- """
- Returns test case complete id.
- """
- return test_case.id()
- def getCaseFile(test_case):
- """
- Returns test case file path.
- """
- return inspect.getsourcefile(test_case.__class__)
- def getCaseMethod(test_case):
- """
- Returns test case method name.
- """
- return getCaseID(test_case).split('.')[-1]
|