39 lines
1.3 KiB
C++
39 lines
1.3 KiB
C++
// Copyright (c) OpenMMLab. All rights reserved.
|
|
|
|
// clang-format off
|
|
#include "catch.hpp"
|
|
// clang-format on
|
|
#include "mmdeploy/core/model.h"
|
|
#include "mmdeploy/core/model_impl.h"
|
|
#include "test_resource.h"
|
|
|
|
using namespace mmdeploy;
|
|
using namespace framework;
|
|
|
|
TEST_CASE("test directory model", "[.model][resource]") {
|
|
std::unique_ptr<ModelImpl> model_impl;
|
|
for (auto& entry : gRegistry<ModelImpl>().Creators()) {
|
|
if (entry->name() == "DirectoryModel") {
|
|
model_impl = entry->Create();
|
|
break;
|
|
}
|
|
}
|
|
REQUIRE(model_impl);
|
|
|
|
auto& gResource = MMDeployTestResources::Get();
|
|
auto directory_model_list = gResource.LocateModelResources("sdk_models");
|
|
REQUIRE(!directory_model_list.empty());
|
|
auto model_dir = "sdk_models/good_model";
|
|
REQUIRE(gResource.IsDir(model_dir));
|
|
auto model_path = gResource.resource_root_path() / model_dir;
|
|
REQUIRE(!model_impl->Init(model_path.string()).has_error());
|
|
REQUIRE(!model_impl->ReadFile("deploy.json").has_error());
|
|
REQUIRE(model_impl->ReadFile("not-existing-file").has_error());
|
|
|
|
model_dir = "sdk_models/bad_model";
|
|
REQUIRE(gResource.IsDir(model_dir));
|
|
model_path = gResource.resource_root_path() / model_dir;
|
|
REQUIRE(!model_impl->Init(model_path.string()).has_error());
|
|
REQUIRE(model_impl->ReadMeta().has_error());
|
|
}
|