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