Update unittest to new message

Accelize · Apr 30, 2020 · d58cb43 · d58cb43
1 parent f35e4e0
commit d58cb43
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_unittest_on_hw.py b/tests/test_unittest_on_hw.py
@@ -2768,7 +2768,7 @@ def my_wrong_read_callback(register_offset, returned_data):
             driver.write_register_callback,
             async_cb.callback
         )
-    assert 'DRM Communication Self-Test 2 failed: Could not access DRM Controller registers' in str(excinfo.value)
+    assert 'DRM Communication Self-Test 2 failed' in str(excinfo.value)
     assert 'Please verify' in str(excinfo.value)
     assert async_handler.get_error_code(str(excinfo.value)) == accelize_drm.exceptions.DRMBadArg.error_code
     async_cb.assert_NoError()
@@ -2784,7 +2784,7 @@ def my_wrong_write_callback(register_offset, data_to_write):
             my_wrong_write_callback,
             async_cb.callback
         )
-    assert 'DRM Communication Self-Test 2 failed: Could not access DRM Controller registers' in str(excinfo.value)
+    assert 'DRM Communication Self-Test 2 failed' in str(excinfo.value)
     assert 'Please verify' in str(excinfo.value)
     assert async_handler.get_error_code(str(excinfo.value)) == accelize_drm.exceptions.DRMBadArg.error_code
     async_cb.assert_NoError()