Skip to content

Commit 5f98245

Browse files
committed
Rewrite multi-line test without YAML
Signed-off-by: John Pennycook <john.pennycook@intel.com>
1 parent 6b07796 commit 5f98245

File tree

2 files changed

+44
-20
lines changed

2 files changed

+44
-20
lines changed

tests/multi_line/multi_line.yaml

Lines changed: 0 additions & 11 deletions
This file was deleted.

tests/multi_line/test_multi_line.py

Lines changed: 44 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,15 @@
1-
# Copyright (C) 2019 Intel Corporation
1+
# Copyright (C) 2019-2024 Intel Corporation
22
# SPDX-License-Identifier: BSD-3-Clause
33

4-
import unittest
54
import logging
6-
from codebasin import config, finder, walkers
5+
import os
6+
import unittest
7+
8+
from codebasin import finder
79
from codebasin.walkers.platform_mapper import PlatformMapper
810

9-
class TestExampleFile(unittest.TestCase):
11+
12+
class TestMultiLine(unittest.TestCase):
1013
"""
1114
Simple test of ability to handle counting of multi-line directives
1215
"""
@@ -15,17 +18,49 @@ def setUp(self):
1518
self.rootdir = "./tests/multi_line/"
1619
logging.getLogger("codebasin").disabled = True
1720

18-
self.expected_setmap = {frozenset([]): 4,
19-
frozenset(['CPU', 'GPU']): 17}
21+
self.expected_setmap = {
22+
frozenset([]): 4,
23+
frozenset(["CPU", "GPU"]): 17,
24+
}
2025

2126
def test_yaml(self):
2227
"""multi_line/multi_line.yaml"""
23-
codebase, configuration = config.load("./tests/multi_line/multi_line.yaml", self.rootdir)
28+
codebase = {
29+
"files": [
30+
os.path.realpath(os.path.join(self.rootdir, "main.cpp")),
31+
],
32+
"platforms": ["CPU", "GPU"],
33+
"exclude_files": set(),
34+
"exclude_patterns": [],
35+
"rootdir": self.rootdir,
36+
}
37+
configuration = {
38+
"CPU": [
39+
{
40+
"file": codebase["files"][0],
41+
"defines": ["CPU"],
42+
"include_paths": [],
43+
"include_files": [],
44+
},
45+
],
46+
"GPU": [
47+
{
48+
"file": codebase["files"][0],
49+
"defines": ["GPU"],
50+
"include_paths": [],
51+
"include_files": [],
52+
},
53+
],
54+
}
2455
state = finder.find(self.rootdir, codebase, configuration)
2556
mapper = PlatformMapper(codebase)
2657
setmap = mapper.walk(state)
27-
self.assertDictEqual(setmap, self.expected_setmap, "Mismatch in setmap")
58+
self.assertDictEqual(
59+
setmap,
60+
self.expected_setmap,
61+
"Mismatch in setmap",
62+
)
2863

2964

30-
if __name__ == '__main__':
65+
if __name__ == "__main__":
3166
unittest.main()

0 commit comments

Comments
 (0)