diff options
| -rwxr-xr-x | scripts/march-to-cpu-opt | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/scripts/march-to-cpu-opt b/scripts/march-to-cpu-opt index 0f3f08d..60d992a 100755 --- a/scripts/march-to-cpu-opt +++ b/scripts/march-to-cpu-opt @@ -2,6 +2,7 @@ import argparse import sys +import unittest EXT_OPTS = { "zba": "zba=true", @@ -129,8 +130,22 @@ def conver_arch_to_qemu_cpu_opt(march): cpu_opt.append(EXT_OPTS[ext]) return ",".join(cpu_opt) + +class TestArchStringParse(unittest.TestCase): + def _test(self, arch, expected_arch_list, expected_vlen=0): + exts = parse_march(arch) + vlen = get_vlen(exts) + self.assertEqual(expected_vlen, vlen) + self.assertEqual(set(expected_arch_list), set(exts.keys())) + + def test_rv64gc(self): + self._test("rv64gc", ['i', 'm', 'a', 'f', 'd', 'c']) + self._test("rv32imc_zve32x", ['i', 'm', 'c', 'zve32x'], expected_vlen=32) + self._test("rv32imc_zve32x_zvl128b", ['i', 'm', 'c', 'zve32x', 'zvl128b'], expected_vlen=128) + + def selftest(): - print(parse_march("rv64gc")) + unittest.main(argv=sys.argv[1:]) def main(argv): opt = parse_opt(argv) |