Add test for is_fully_supported

nv-morpheus · May 1, 2024 · 3931259 · 3931259
1 parent c49d3a7
commit 3931259
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/morpheus/_lib/tests/objects/test_dtype.cpp b/morpheus/_lib/tests/objects/test_dtype.cpp
@@ -283,4 +283,17 @@ TEST_F(TestDType, FromCudfNotSupported)
     EXPECT_THROW(DType::from_cudf(cudf::type_id::DECIMAL128), std::invalid_argument);
     EXPECT_THROW(DType::from_cudf(cudf::type_id::STRUCT), std::invalid_argument);
     EXPECT_THROW(DType::from_cudf(cudf::type_id::NUM_TYPE_IDS), std::invalid_argument);
-}
+}
+
+TEST_F(TestDType, IsFullySupported)
+{
+    std::set<TypeId> unsupported_types = {TypeId::EMPTY, TypeId::STRING, TypeId::NUM_TYPE_IDS};
+    for (auto type_id = static_cast<int32_t>(TypeId::EMPTY); type_id <= static_cast<int32_t>(TypeId::NUM_TYPE_IDS);
+         ++type_id)
+    {
+        auto enum_type_id = static_cast<TypeId>(type_id);
+        auto dtype        = DType(enum_type_id);
+
+        ASSERT_EQ(dtype.is_fully_supported(), !unsupported_types.contains(enum_type_id));
+    }
+}