| 1460 | class FilterNetTest : public ::testing::Test { |
| 1461 | protected: |
| 1462 | void RunFilterNetTest( |
| 1463 | const string& input_param_string, const string& filtered_param_string) { |
| 1464 | NetParameter input_param; |
| 1465 | CHECK(google::protobuf::TextFormat::ParseFromString( |
| 1466 | input_param_string, &input_param)); |
| 1467 | NetParameter expected_filtered_param; |
| 1468 | CHECK(google::protobuf::TextFormat::ParseFromString( |
| 1469 | filtered_param_string, &expected_filtered_param)); |
| 1470 | NetParameter actual_filtered_param; |
| 1471 | Net<float>::FilterNet(input_param, &actual_filtered_param); |
| 1472 | EXPECT_EQ(expected_filtered_param.DebugString(), |
| 1473 | actual_filtered_param.DebugString()); |
| 1474 | // Also test idempotence. |
| 1475 | NetParameter double_filtered_param; |
| 1476 | Net<float>::FilterNet(actual_filtered_param, &double_filtered_param); |
| 1477 | EXPECT_EQ(actual_filtered_param.DebugString(), |
| 1478 | double_filtered_param.DebugString()); |
| 1479 | } |
| 1480 | }; |
| 1481 | |
| 1482 | TEST_F(FilterNetTest, TestNoFilter) { |