Tests: Refactor payload_to_info()

[vpp.git] / test / test_vom.py

diff --git a/test/test_vom.py b/test/test_vom.py
index 6aba555..a27d531 100644 (file)
--- a/test/test_vom.py
+++ b/test/test_vom.py
@@ -8,10 +8,18 @@ from framework import VppTestCase, running_extended_tests, \
     VppTestRunner, Worker
 
 
-@unittest.skipUnless(running_extended_tests(), "part of extended tests")
+@unittest.skipUnless(running_extended_tests, "part of extended tests")
 class VOMTestCase(VppTestCase):
     """ VPP Object Model Test """
 
+    @classmethod
+    def setUpClass(cls):
+        super(VOMTestCase, cls).setUpClass()
+
+    @classmethod
+    def tearDownClass(cls):
+        super(VOMTestCase, cls).tearDownClass()
+
     def test_vom_cpp(self):
         """ run C++ VOM tests """
         var = "TEST_DIR"