| 80 | class SliceOpTest : public ::testing::TestWithParam<TestType> {}; |
| 81 | |
| 82 | TEST_P(SliceOpTest, In1D) { |
| 83 | SliceOpModel<float, int32_t> m({4}, {1}, {1}, {1}, {2}, TensorType_INT32, |
| 84 | TensorType_FLOAT32, GetParam()); |
| 85 | m.SetInput({1, 2, 3, 4}); |
| 86 | m.Invoke(); |
| 87 | EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); |
| 88 | EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3})); |
| 89 | } |
| 90 | |
| 91 | TEST_P(SliceOpTest, In2D) { |
| 92 | SliceOpModel<float, int32_t> m({2, 3}, {2}, {1, 0}, {2}, {1, 2}, |
nothing calls this directly
no test coverage detected