def check_counting_condition(self, conditions, function, expected):
"""
Checks if the given count vectors match the given expected data.
:param conditions: String indicating which condition(s) to use (will
be fed to
ClangCountingConditions.counting_condition)
:param function: String indicating which function from test file to
use. (i.e. "used(int, int)")
:param expected: Dict with python lists of counts for all variables.
"""
counter = ClangCountVectorCreator(
ClangCountingConditions.counting_condition(
Setting("irrelevant", conditions)))
vectors = counter.get_vectors_for_file(self.testfile)
actual = vectors[function]
self.assertEqual(len(actual),
len(expected),
"Actual dict: " + str(actual))
self.assertEqual(sorted(actual.keys()),
sorted(expected.keys()))
for variable in actual:
self.assertEqual(actual[variable].count_vector,
expected[variable],
"Variable '{}' doesnt match.".format(variable))
class ClangCountVectorCreatorTest(unittest.TestCase):
functions = sorted(["main(int, char *)", "test()"])
def setUp(self):
self.testfile = os.path.abspath(os.path.join(
os.path.dirname(__file__),
"sample.c"))
def test_empty_counting(self):
expected_results = {
(6, "test()"): {},
(12, "main(int, char *)"): {
# Variables
"i": [],
"asd": [],
"t": [],
"args": [],
# Globals
"g": [],
# Functions
"smile": [],
"printf": [],
# Constants
"#5": [],
'#"i is %d"': []}}
self.uut = ClangCountVectorCreator()
cv_dict = self.uut.get_vectors_for_file(self.testfile)
self.check_cv_dict(cv_dict, expected_results)
def check_cv_dict(self, actual, expected):
self.assertEqual(len(actual), len(expected), str(actual))
self.assertEqual(sorted(actual.keys()), sorted(expected.keys()))
for function in actual:
self.assertEqual(len(actual[function]), len(expected[function]))
self.assertEqual(sorted(actual[function].keys()),
sorted(expected[function].keys()))
for variable in actual[function]:
self.assertEqual(actual[function][variable].count_vector,
expected[function][variable])
def test_counting(self):
expected_results = {
(6, "test()"): {},
(12, "main(int, char *)"): {
# Variables
"i": [4, 1],
"asd": [1, 0],
"t": [4, 1],
"args": [2, 0],
# Globals
"g": [3, 1],
# Functions
"smile": [1, 1],
"printf": [1, 1],
# Constants
"#5": [1, 0],
'#"i is %d"': [1, 1]}}
self.uut = ClangCountVectorCreator([no_condition, is_call_argument])
cv_dict = self.uut.get_vectors_for_file(self.testfile)
self.check_cv_dict(cv_dict, expected_results)
