vlib: improve code coverage in src/vlib

[vpp.git] / test / test_fib.py

diff --git a/test/test_fib.py b/test/test_fib.py
index 6b59746..faafdad 100644 (file)
--- a/test/test_fib.py
+++ b/test/test_fib.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 
 import unittest
 
@@ -8,13 +8,21 @@ from framework import VppTestCase, VppTestRunner
 class TestFIB(VppTestCase):
     """ FIB Test Case """
 
+    @classmethod
+    def setUpClass(cls):
+        super(TestFIB, cls).setUpClass()
+
+    @classmethod
+    def tearDownClass(cls):
+        super(TestFIB, cls).tearDownClass()
+
     def test_fib(self):
         """ FIB Unit Tests """
         error = self.vapi.cli("test fib")
 
         if error:
             self.logger.critical(error)
-        self.assertEqual(error.find("Failed"), -1)
+        self.assertNotIn("Failed", error)
 
 if __name__ == '__main__':
     unittest.main(testRunner=VppTestRunner)