diff options
Diffstat (limited to 'test/test_node_variants.py')
| -rw-r--r-- | test/test_node_variants.py | 40 |
1 files changed, 22 insertions, 18 deletions
diff --git a/test/test_node_variants.py b/test/test_node_variants.py index 3632910e3cb..a5c9137f9ba 100644 --- a/test/test_node_variants.py +++ b/test/test_node_variants.py @@ -13,15 +13,14 @@ def skipVariant(variant): with open("/proc/cpuinfo") as f: cpuinfo = f.read() - exp = re.compile( - r'(?:flags\s+:)(?:\s\w+)+(?:\s(' + variant + r'))(?:\s\w+)+') + exp = re.compile(r"(?:flags\s+:)(?:\s\w+)+(?:\s(" + variant + r"))(?:\s\w+)+") match = exp.search(cpuinfo, re.DOTALL | re.MULTILINE) return checkX86() and match is not None class TestNodeVariant(VppTestCase): - """ Test Node Variants """ + """Test Node Variants""" @classmethod def setUpConstants(cls, variant): @@ -30,9 +29,10 @@ class TestNodeVariant(VppTestCase): if checkX86(): node_variants = cls.vpp_cmdline.index("node { ") + 1 - cls.vpp_cmdline[node_variants] = ("default { variant default } " - "ip4-rewrite { variant " + - variant + " } ") + cls.vpp_cmdline[node_variants] = ( + "default { variant default } " + "ip4-rewrite { variant " + variant + " } " + ) @classmethod def setUpClass(cls): @@ -52,13 +52,14 @@ class TestNodeVariant(VppTestCase): node_desc = self.vapi.cli("show node " + node) self.logger.info(node_desc) - match = re.search(r'\s+(\S+)\s+(\d+)\s+(:?yes)', - node_desc, re.DOTALL | re.MULTILINE) + match = re.search( + r"\s+(\S+)\s+(\d+)\s+(:?yes)", node_desc, re.DOTALL | re.MULTILINE + ) return match.groups(0) def checkVariant(self, variant): - """ Test node variants defaults """ + """Test node variants defaults""" variant_info = self.getActiveVariant("ip4-lookup") self.assertEqual(variant_info[0], "default") @@ -68,7 +69,7 @@ class TestNodeVariant(VppTestCase): class TestICLVariant(TestNodeVariant): - """ Test icl Node Variants """ + """Test icl Node Variants""" VARIANT = "icl" LINUX_VARIANT = "avx512_bitalg" @@ -85,14 +86,15 @@ class TestICLVariant(TestNodeVariant): def tearDownClass(cls): super(TestICLVariant, cls).tearDownClass() - @unittest.skipUnless(skipVariant(LINUX_VARIANT), - VARIANT + " not a supported variant, skip.") + @unittest.skipUnless( + skipVariant(LINUX_VARIANT), VARIANT + " not a supported variant, skip." + ) def test_icl(self): self.checkVariant(self.VARIANT) class TestSKXVariant(TestNodeVariant): - """ Test skx Node Variants """ + """Test skx Node Variants""" VARIANT = "skx" LINUX_VARIANT = "avx512f" @@ -109,14 +111,15 @@ class TestSKXVariant(TestNodeVariant): def tearDownClass(cls): super(TestSKXVariant, cls).tearDownClass() - @unittest.skipUnless(skipVariant(LINUX_VARIANT), - VARIANT + " not a supported variant, skip.") + @unittest.skipUnless( + skipVariant(LINUX_VARIANT), VARIANT + " not a supported variant, skip." + ) def test_skx(self): self.checkVariant(self.VARIANT) class TestHSWVariant(TestNodeVariant): - """ Test avx2 Node Variants """ + """Test avx2 Node Variants""" VARIANT = "hsw" LINUX_VARIANT = "avx2" @@ -133,7 +136,8 @@ class TestHSWVariant(TestNodeVariant): def tearDownClass(cls): super(TestHSWVariant, cls).tearDownClass() - @unittest.skipUnless(skipVariant(LINUX_VARIANT), - VARIANT + " not a supported variant, skip.") + @unittest.skipUnless( + skipVariant(LINUX_VARIANT), VARIANT + " not a supported variant, skip." + ) def test_hsw(self): self.checkVariant(self.VARIANT) |