TEST/UCP/DEVICE: Improve error detection

Author: yosefe

test/gtest/ucp/test_ucp_device.cc

@@ -124,7 +124,6 @@ UCS_TEST_P(test_ucp_device, mapped_buffer_kernel_memcmp)
 
 UCS_TEST_P(test_ucp_device, create_success)
 {
-    scoped_log_handler wrap_err(wrap_errors_logger);
     mem_list list(sender(), receiver());
     ucp_device_mem_list_handle_h handle = nullptr;
 
@@ -172,6 +171,8 @@ UCS_TEST_P(test_ucp_device, create_fail)
     ucp_device_mem_list_handle_h handle       = nullptr;
     ucp_device_mem_list_params_t empty_params = {};
 
+    scoped_log_handler wrap_err(wrap_errors_logger);
+
     ASSERT_EQ(UCS_ERR_INVALID_PARAM,
               ucp_device_mem_list_create(sender().ep(), NULL, &handle));
 

test/gtest/uct/test_device.cc

@@ -28,20 +28,15 @@ class test_device : public uct_test {
         uct_test::init();
         status = uct_cuda_base_get_cuda_device(GetParam()->sys_device,
                                                &m_cuda_dev);
-        if (status != UCS_OK) {
-            return;
-        }
+        ASSERT_UCS_OK(status, << " sys_device "
+                              << static_cast<int>(GetParam()->sys_device));
 
         status = UCT_CUDADRV_FUNC_LOG_ERR(
                 cuDevicePrimaryCtxRetain(&ctx, m_cuda_dev));
-        if (status != UCS_OK) {
-            return;
-        }
+        ASSERT_UCS_OK(status);
 
         status = UCT_CUDADRV_FUNC_LOG_ERR(cuCtxPushCurrent(ctx));
-        if (status != UCS_OK) {
-            return;
-        }
+        ASSERT_UCS_OK(status);
 
         m_receiver = uct_test::create_entity(0);
         m_entities.push_back(m_receiver);