Clean up pytorch_android_torchvision test (#29455)

dreiss · facebook-github-bot · commit 0cfa4965a24a · 2019-11-08T14:29:10.000-08:00
Summary: Pull Request resolved: #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
diff --git a/android/pytorch_android_torchvision/src/androidTest/java/org/pytorch/torchvision/TorchVisionInstrumentedTests.java b/android/pytorch_android_torchvision/src/androidTest/java/org/pytorch/torchvision/TorchVisionInstrumentedTests.java
@@ -14,11 +14,6 @@
 @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 void smokeTest() {
             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());
   }
 }
