talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Project import generated by Copybara.

Browse Source 
GitOrigin-RevId: d073f8e21be2fcc0e503cb97c6695078b6b75310
This commit is contained in:
MediaPipe Team
2021-02-27 03:30:05 -05:00
committed by chuoling
parent 39309bedba
commit 350fbb2100
755 changed files with 16391 additions and 11075 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/framework/calculator_base_test.cc
+20 -26
View File
@@ -41,7 +41,7 @@ namespace test_ns {
// streams and input side packets.
class DeadEndCalculator : public CalculatorBase {
public:
static mediapipe::Status GetContract(CalculatorContract* cc) {
static absl::Status GetContract(CalculatorContract* cc) {
for (int i = 0; i < cc->Inputs().NumEntries(); ++i) {
cc->Inputs().Index(i).SetAny();
}
@@ -51,16 +51,14 @@ class DeadEndCalculator : public CalculatorBase {
for (int i = 0; i < cc->InputSidePackets().NumEntries(); ++i) {
cc->InputSidePackets().Index(i).SetAny();
}
return mediapipe::OkStatus();
return absl::OkStatus();
}
mediapipe::Status Open(CalculatorContext* cc) override {
return mediapipe::OkStatus();
}
absl::Status Open(CalculatorContext* cc) override { return absl::OkStatus(); }
mediapipe::Status Process(CalculatorContext* cc) override {
absl::Status Process(CalculatorContext* cc) override {
if (cc->Inputs().NumEntries() > 0) {
return mediapipe::OkStatus();
return absl::OkStatus();
} else {
// This is a source calculator, but we don't produce any outputs.
return tool::StatusStop();
@@ -73,14 +71,12 @@ namespace whitelisted_ns {
class DeadCalculator : public CalculatorBase {
public:
static mediapipe::Status GetContract(CalculatorContract* cc) {
return mediapipe::OkStatus();
static absl::Status GetContract(CalculatorContract* cc) {
return absl::OkStatus();
}
mediapipe::Status Open(CalculatorContext* cc) override {
return mediapipe::OkStatus();
}
mediapipe::Status Process(CalculatorContext* cc) override {
return mediapipe::OkStatus();
absl::Status Open(CalculatorContext* cc) override { return absl::OkStatus(); }
absl::Status Process(CalculatorContext* cc) override {
return absl::OkStatus();
}
};
@@ -89,14 +85,12 @@ class DeadCalculator : public CalculatorBase {
class EndCalculator : public CalculatorBase {
public:
static mediapipe::Status GetContract(CalculatorContract* cc) {
return mediapipe::OkStatus();
static absl::Status GetContract(CalculatorContract* cc) {
return absl::OkStatus();
}
mediapipe::Status Open(CalculatorContext* cc) override {
return mediapipe::OkStatus();
}
mediapipe::Status Process(CalculatorContext* cc) override {
return mediapipe::OkStatus();
absl::Status Open(CalculatorContext* cc) override { return absl::OkStatus(); }
absl::Status Process(CalculatorContext* cc) override {
return absl::OkStatus();
}
};
REGISTER_CALCULATOR(::mediapipe::EndCalculator);
@@ -105,7 +99,7 @@ namespace {
TEST(CalculatorTest, SourceProcessOrder) {
internal::Collection<OutputStreamManager> output_stream_managers(
tool::CreateTagMap(2).ValueOrDie());
tool::CreateTagMap(2).value());
PacketType output0_type;
PacketType output1_type;
@@ -117,7 +111,7 @@ TEST(CalculatorTest, SourceProcessOrder) {
MP_ASSERT_OK(
output_stream_managers.Index(1).Initialize("output1", &output1_type));
PacketSet input_side_packets(tool::CreateTagMap({}).ValueOrDie());
PacketSet input_side_packets(tool::CreateTagMap({}).value());
CalculatorState calculator_state("Node", /*node_id=*/0, "Calculator",
CalculatorGraphConfig::Node(), nullptr);
@@ -126,7 +120,7 @@ TEST(CalculatorTest, SourceProcessOrder) {
CalculatorContextManager calculator_context_manager;
CalculatorContext calculator_context(&calculator_state,
tool::CreateTagMap({}).ValueOrDie(),
tool::CreateTagMap({}).value(),
output_stream_managers.TagMap());
OutputStreamShardSet& output_set = calculator_context.Outputs();
output_set.Index(0).SetSpec(output_stream_managers.Index(0).Spec());
@@ -167,13 +161,13 @@ TEST(CalculatorTest, CreateByName) {
"mediapipe", "DeadEndCalculator")
.status()
.code(),
mediapipe::StatusCode::kNotFound);
absl::StatusCode::kNotFound);
EXPECT_EQ(CalculatorBaseRegistry::CreateByName( //
"DeadEndCalculator")
.status()
.code(),
mediapipe::StatusCode::kNotFound);
absl::StatusCode::kNotFound);
}
// Tests registration of a calculator within a whitelisted namespace.