Function main

tests/unittests/test_zipenhancer_utility.cpp:95–127

}  // namespace

int main() {
    try {
        const auto model = engine::audio::ZipEnhancerModel::load_from_directory(
            repo_path("assets/framework/audio_utilities/zipenhancer"),
            cuda_backend());
        const auto fixture_dir = asset_path("framework/audio_utilities/zipenhancer");
        std::vector<int> cases;
        for (const auto & entry : std::filesystem::directory_iterator(fixture_dir)) {
            const auto name = entry.path().filename().string();
            constexpr const char * prefix = "zipenhancer_case";
            constexpr const char * suffix = ".safetensors";
            if (name.rfind(prefix, 0) != 0 ||
                name.size() <= std::char_traits<char>::length(prefix) + std::char_traits<char>::length(suffix) ||
                name.substr(name.size() - std::char_traits<char>::length(suffix)) != suffix) {
                continue;
            }
            const auto index = name.substr(
                std::char_traits<char>::length(prefix),
                name.size() - std::char_traits<char>::length(prefix) - std::char_traits<char>::length(suffix));
            cases.push_back(std::stoi(index));
        }
        std::sort(cases.begin(), cases.end());
        require(cases.size() >= 2, "ZipEnhancer requires at least two reference fixtures");
        for (const int case_index : cases) {
            run_case(model, case_index);
        }
        std::cout << "zipenhancer_utility_test passed\n";
    } catch (const std::exception & ex) {
        std::cerr << "zipenhancer_utility_test failed: " << ex.what() << "\n";
        return 1;
    }
    return 0;
}