X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=tests%2Fbindings%2Fpython%2Fbt2%2Ftest_trace_class.py;fp=tests%2Fbindings%2Fpython%2Fbt2%2Ftest_trace_class.py;h=4f19b4a2bc8c173d9986ab68119a93600fe232b8;hb=4430bc809d78701c128ced642fb4e9597bc00ad0;hp=265b0de8b40ed8ac0604216688235a870a7a8a95;hpb=7e4b80cc148d07716d605eb220f6f54aeddd9363;p=babeltrace.git

diff --git a/tests/bindings/python/bt2/test_trace_class.py b/tests/bindings/python/bt2/test_trace_class.py
index 265b0de8..4f19b4a2 100644
--- a/tests/bindings/python/bt2/test_trace_class.py
+++ b/tests/bindings/python/bt2/test_trace_class.py
@@ -45,7 +45,7 @@ class TraceClassTestCase(unittest.TestCase):
         tc = run_in_component_init(f)
         self.assertTrue(tc.assigns_automatic_stream_class_id)
 
-        with self.assertRaises(bt2.CreationError):
+        with self.assertRaises(ValueError):
             sc1 = tc.create_stream_class(23)
 
     def test_no_assigns_automatic_stream_class_id(self):
@@ -66,7 +66,7 @@ class TraceClassTestCase(unittest.TestCase):
         self.assertFalse(tc.assigns_automatic_stream_class_id)
 
         # In this mode, it is required to pass an explicit id.
-        with self.assertRaises(bt2.CreationError):
+        with self.assertRaises(ValueError):
             tc.create_stream_class()
 
     def test_env_get(self):