test: improve coverage of negative tests for null generators (#519)

ROCm · Aug 15, 2024 · 76d16a5 · 76d16a5
1 parent 29992dd
commit 76d16a5
Showing 5 changed files with 65 additions and 37 deletions.
diff --git a/test/test_rocrand_generate.cpp b/test/test_rocrand_generate.cpp
@@ -140,13 +140,21 @@ TEST_P(rocrand_generate_tests, short_test)
 TEST(rocrand_generate_tests, neg_test)
 {
     const size_t size = 256;
-    unsigned int * data = NULL;
+    void*        data = nullptr;
 
-    rocrand_generator generator = NULL;
-    EXPECT_EQ(
-        rocrand_generate(generator, (unsigned int *) data, size),
-        ROCRAND_STATUS_NOT_CREATED
-    );
+    rocrand_generator generator = nullptr;
+
+    EXPECT_EQ(rocrand_generate(generator, static_cast<unsigned int*>(data), size),
+              ROCRAND_STATUS_NOT_CREATED);
+
+    EXPECT_EQ(rocrand_generate_char(generator, static_cast<unsigned char*>(data), size),
+              ROCRAND_STATUS_NOT_CREATED);
+
+    EXPECT_EQ(rocrand_generate_short(generator, static_cast<unsigned short*>(data), size),
+              ROCRAND_STATUS_NOT_CREATED);
+
+    EXPECT_EQ(rocrand_generate_long_long(generator, static_cast<unsigned long long*>(data), size),
+              ROCRAND_STATUS_NOT_CREATED);
 }
 
 INSTANTIATE_TEST_SUITE_P(rocrand_generate_tests,

diff --git a/test/test_rocrand_generate_log_normal.cpp b/test/test_rocrand_generate_log_normal.cpp
@@ -145,16 +145,26 @@ TEST_P(rocrand_generate_log_normal_tests, half_test)
 
 TEST(rocrand_generate_log_normal_tests, neg_test)
 {
-    const size_t size = 256;
-    float mean = 5.0;
-    float stddev = 2.0;
-    float * data = NULL;
+    const size_t size   = 256;
+    float        mean   = 5.0;
+    float        stddev = 2.0;
+    void*        data   = nullptr;
+
+    rocrand_generator generator = nullptr;
+
+    EXPECT_EQ(rocrand_generate_log_normal(generator, static_cast<float*>(data), size, mean, stddev),
+              ROCRAND_STATUS_NOT_CREATED);
+
+    EXPECT_EQ(rocrand_generate_log_normal_double(generator,
+                                                 static_cast<double*>(data),
+                                                 size,
+                                                 mean,
+                                                 stddev),
+              ROCRAND_STATUS_NOT_CREATED);
 
-    rocrand_generator generator = NULL;
     EXPECT_EQ(
-        rocrand_generate_log_normal(generator, (float *) data, size, mean, stddev),
-        ROCRAND_STATUS_NOT_CREATED
-    );
+        rocrand_generate_log_normal_half(generator, static_cast<half*>(data), size, mean, stddev),
+        ROCRAND_STATUS_NOT_CREATED);
 }
 
 INSTANTIATE_TEST_SUITE_P(rocrand_generate_log_normal_tests,

diff --git a/test/test_rocrand_generate_normal.cpp b/test/test_rocrand_generate_normal.cpp
@@ -145,16 +145,22 @@ TEST_P(rocrand_generate_normal_tests, half_test)
 
 TEST(rocrand_generate_normal_tests, neg_test)
 {
-    const size_t size = 256;
-    float mean = 5.0;
-    float stddev = 2.0;
-    float * data = NULL;
+    const size_t size   = 256;
+    float        mean   = 5.0;
+    float        stddev = 2.0;
+    void*        data   = nullptr;
+
+    rocrand_generator generator = nullptr;
+
+    EXPECT_EQ(rocrand_generate_normal(generator, static_cast<float*>(data), size, mean, stddev),
+              ROCRAND_STATUS_NOT_CREATED);
 
-    rocrand_generator generator = NULL;
     EXPECT_EQ(
-        rocrand_generate_normal(generator, (float *) data, size, mean, stddev),
-        ROCRAND_STATUS_NOT_CREATED
-    );
+        rocrand_generate_normal_double(generator, static_cast<double*>(data), size, mean, stddev),
+        ROCRAND_STATUS_NOT_CREATED);
+
+    EXPECT_EQ(rocrand_generate_normal_half(generator, static_cast<half*>(data), size, mean, stddev),
+              ROCRAND_STATUS_NOT_CREATED);
 }
 
 INSTANTIATE_TEST_SUITE_P(rocrand_generate_normal_tests,

diff --git a/test/test_rocrand_generate_poisson.cpp b/test/test_rocrand_generate_poisson.cpp
@@ -147,15 +147,14 @@ TEST_P(rocrand_generate_poisson_tests, generate_test)
 
 TEST(rocrand_generate_poisson_tests, neg_test)
 {
-    const size_t size = 256;
-    double         lambda = 100.0;
-    unsigned int * data = NULL;
-
-    rocrand_generator generator = NULL;
-    EXPECT_EQ(
-        rocrand_generate_poisson(generator, (unsigned int *)data, size, lambda),
-        ROCRAND_STATUS_NOT_CREATED
-    );
+    const size_t size   = 256;
+    double       lambda = 100.0;
+    void*        data   = nullptr;
+
+    rocrand_generator generator = nullptr;
+
+    EXPECT_EQ(rocrand_generate_poisson(generator, static_cast<unsigned int*>(data), size, lambda),
+              ROCRAND_STATUS_NOT_CREATED);
 }
 
 TEST_P(rocrand_generate_poisson_tests, out_of_range_test)

diff --git a/test/test_rocrand_generate_uniform.cpp b/test/test_rocrand_generate_uniform.cpp
@@ -140,13 +140,18 @@ TEST_P(rocrand_generate_uniform_tests, half_test)
 TEST(rocrand_generate_uniform_tests, neg_test)
 {
     const size_t size = 256;
-    float * data = NULL;
+    void*        data = nullptr;
 
-    rocrand_generator generator = NULL;
-    EXPECT_EQ(
-        rocrand_generate_uniform(generator, (float *) data, size),
-        ROCRAND_STATUS_NOT_CREATED
-    );
+    rocrand_generator generator = nullptr;
+
+    EXPECT_EQ(rocrand_generate_uniform(generator, static_cast<float*>(data), size),
+              ROCRAND_STATUS_NOT_CREATED);
+
+    EXPECT_EQ(rocrand_generate_uniform_double(generator, static_cast<double*>(data), size),
+              ROCRAND_STATUS_NOT_CREATED);
+
+    EXPECT_EQ(rocrand_generate_uniform_half(generator, static_cast<half*>(data), size),
+              ROCRAND_STATUS_NOT_CREATED);
 }
 
 INSTANTIATE_TEST_SUITE_P(rocrand_generate_uniform_tests,