apoorva
aa265edfb1
git-subtree-dir: darknet_ros2 git-subtree-mainline: f33742e6e8ae5b436aa1ce48d8c40cb5e5e5562e git-subtree-split: be4fd04c0ea8fbed80b3549283701e16145422c6
68 lines
1.8 KiB
C++
68 lines
1.8 KiB
C++
// Copyright 2019 Open Source Robotics Foundation, Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "openrobotics_darknet_ros/parse.hpp"
|
|
|
|
using openrobotics::darknet_ros::parse_class_names;
|
|
|
|
TEST(parse, names_file_does_not_exist)
|
|
{
|
|
std::vector<std::string> class_names;
|
|
|
|
class_names = parse_class_names("openrobotics_darknet_ros_file_does_not_exist");
|
|
|
|
ASSERT_EQ(0u, class_names.size());
|
|
}
|
|
|
|
TEST(parse, one_name)
|
|
{
|
|
std::vector<std::string> class_names;
|
|
|
|
class_names = parse_class_names("data/1_class_name.txt");
|
|
|
|
ASSERT_EQ(1u, class_names.size());
|
|
EXPECT_EQ("tomato", class_names[0]);
|
|
}
|
|
|
|
TEST(parse, 3_class_names_with_whitespace)
|
|
{
|
|
std::vector<std::string> class_names;
|
|
|
|
class_names = parse_class_names("data/3_class_names_with_whitespace.txt");
|
|
|
|
ASSERT_EQ(3u, class_names.size());
|
|
EXPECT_EQ("foo", class_names[0]);
|
|
EXPECT_EQ("bar", class_names[1]);
|
|
EXPECT_EQ("baz", class_names[2]);
|
|
}
|
|
|
|
TEST(parse, five_names)
|
|
{
|
|
std::vector<std::string> class_names;
|
|
|
|
class_names = parse_class_names("data/5_class_names.txt");
|
|
|
|
ASSERT_EQ(5u, class_names.size());
|
|
EXPECT_EQ("car", class_names[0]);
|
|
EXPECT_EQ("boat", class_names[1]);
|
|
EXPECT_EQ("bus", class_names[2]);
|
|
EXPECT_EQ("airplane", class_names[3]);
|
|
EXPECT_EQ("space ship", class_names[4]);
|
|
}
|