MediaPipe TeamandSebastian Schmidt
c688862570
GitOrigin-RevId: 6e5aa035cd1f6a9333962df5d3ab97a05bd5744e
57 lines
1.5 KiB
C++
57 lines
1.5 KiB
C++
#include "mediapipe/framework/api2/port.h"
|
|
|
|
#include "mediapipe/framework/port/gtest.h"
|
|
|
|
namespace mediapipe {
|
|
namespace api2 {
|
|
namespace {
|
|
|
|
TEST(PortTest, IntInput) {
|
|
static constexpr auto port = Input<int>("FOO");
|
|
EXPECT_EQ(port.type_id(), kTypeId<int>);
|
|
}
|
|
|
|
TEST(PortTest, OptionalInput) {
|
|
static constexpr auto port = Input<float>::Optional("BAR");
|
|
EXPECT_TRUE(port.IsOptional());
|
|
}
|
|
|
|
TEST(PortTest, Tag) {
|
|
static constexpr auto port = Input<int>("FOO");
|
|
EXPECT_EQ(std::string(port.Tag()), "FOO");
|
|
}
|
|
|
|
struct DeletedCopyType {
|
|
DeletedCopyType(const DeletedCopyType&) = delete;
|
|
DeletedCopyType& operator=(const DeletedCopyType&) = delete;
|
|
};
|
|
|
|
TEST(PortTest, DeletedCopyConstructorInput) {
|
|
static constexpr Input<DeletedCopyType> kInputPort{"INPUT"};
|
|
EXPECT_EQ(std::string(kInputPort.Tag()), "INPUT");
|
|
|
|
static constexpr Output<DeletedCopyType> kOutputPort{"OUTPUT"};
|
|
EXPECT_EQ(std::string(kOutputPort.Tag()), "OUTPUT");
|
|
|
|
static constexpr SideInput<DeletedCopyType> kSideInputPort{"SIDE_INPUT"};
|
|
EXPECT_EQ(std::string(kSideInputPort.Tag()), "SIDE_INPUT");
|
|
|
|
static constexpr SideOutput<DeletedCopyType> kSideOutputPort{"SIDE_OUTPUT"};
|
|
EXPECT_EQ(std::string(kSideOutputPort.Tag()), "SIDE_OUTPUT");
|
|
}
|
|
|
|
class AbstractBase {
|
|
public:
|
|
virtual ~AbstractBase() = default;
|
|
virtual absl::string_view name() const = 0;
|
|
};
|
|
|
|
TEST(PortTest, Abstract) {
|
|
static constexpr Input<AbstractBase> kInputPort{"INPUT"};
|
|
EXPECT_EQ(std::string(kInputPort.Tag()), "INPUT");
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace api2
|
|
} // namespace mediapipe
|