Skip to content

Commit a1805d1

Browse files
committed
Cpp tests
1 parent 24d0035 commit a1805d1

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/decoders/VideoDecoderTest.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ TEST(VideoDecoderTest, RespectsWidthAndHeightFromOptions) {
150150
videoStreamOptions.width = 100;
151151
videoStreamOptions.height = 120;
152152
decoder->addVideoStreamDecoder(-1, videoStreamOptions);
153-
torch::Tensor tensor = decoder->getNextFrameNoDemux().frame;
153+
torch::Tensor tensor = decoder->getNextFrameNoDemux().data;
154154
EXPECT_EQ(tensor.sizes(), std::vector<long>({3, 120, 100}));
155155
}
156156

@@ -161,7 +161,7 @@ TEST(VideoDecoderTest, RespectsOutputTensorDimensionOrderFromOptions) {
161161
VideoDecoder::VideoStreamOptions videoStreamOptions;
162162
videoStreamOptions.dimensionOrder = "NHWC";
163163
decoder->addVideoStreamDecoder(-1, videoStreamOptions);
164-
torch::Tensor tensor = decoder->getNextFrameNoDemux().frame;
164+
torch::Tensor tensor = decoder->getNextFrameNoDemux().data;
165165
EXPECT_EQ(tensor.sizes(), std::vector<long>({270, 480, 3}));
166166
}
167167

@@ -235,7 +235,7 @@ TEST_P(VideoDecoderTest, DecodesFramesInABatchInNHWC) {
235235
VideoDecoder::VideoStreamOptions("dimension_order=NHWC"));
236236
// Frame with index 180 corresponds to timestamp 6.006.
237237
auto output = ourDecoder->getFramesAtIndices(bestVideoStreamIndex, {0, 180});
238-
auto tensor = output.;
238+
auto tensor = output.data;
239239
EXPECT_EQ(tensor.sizes(), std::vector<long>({2, 270, 480, 3}));
240240

241241
torch::Tensor tensor0FromFFMPEG =

0 commit comments

Comments
 (0)