mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-08 01:58:34 +08:00
16 lines
479 B
C++
16 lines
479 B
C++
#include "scheduler/ResourceFactory.h"
|
|
#include <gtest/gtest.h>
|
|
|
|
|
|
using namespace zilliz::milvus::engine;
|
|
|
|
TEST(resource_factory_test, create) {
|
|
auto disk = ResourceFactory::Create("disk");
|
|
auto cpu = ResourceFactory::Create("cpu");
|
|
auto gpu = ResourceFactory::Create("gpu");
|
|
|
|
ASSERT_TRUE(std::dynamic_pointer_cast<DiskResource>(disk));
|
|
ASSERT_TRUE(std::dynamic_pointer_cast<CpuResource>(cpu));
|
|
ASSERT_TRUE(std::dynamic_pointer_cast<GpuResource>(gpu));
|
|
}
|