OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Clean up pytorch_android_torchvision test (#29455)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/29455

- Don't need to load native library.
- Shape is now private.

Test Plan: Ran test.

Reviewed By: IvanKobzarev

Differential Revision: D18405213

fbshipit-source-id: e1d1abcf2122332317693ce391e840904b69e135
This commit is contained in:
David Reiss 2019-11-08 14:25:03 -08:00 committed by Facebook Github Bot
parent abf55eb3a8
commit 0cfa4965a2
1 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
android/pytorch_android_torchvision/src/androidTest/java/org/pytorch/torchvision/TorchVisionInstrumentedTests.java
View File

@ -14,11 +14,6 @@ import static org.junit.Assert.assertArrayEquals;
@RunWith(AndroidJUnit4.class)
public class TorchVisionInstrumentedTests {
@Before
public void setUp() {
System.loadLibrary("pytorch");
}
@Test
public void smokeTest() {
Bitmap bitmap = Bitmap.createBitmap(320, 240, Bitmap.Config.ARGB_8888);
@ -27,6 +22,6 @@ public class TorchVisionInstrumentedTests {
bitmap,
TensorImageUtils.TORCHVISION_NORM_MEAN_RGB,
TensorImageUtils.TORCHVISION_NORM_STD_RGB);
assertArrayEquals(new long[] {1l, 3l, 240l, 320l}, tensor.shape);
assertArrayEquals(new long[] {1l, 3l, 240l, 320l}, tensor.shape());
}
}