Fix equality check and star import

Author: karllessard

ndarray/src/main/java/org/tensorflow/ndarray/impl/dense/AbstractDenseNdArray.java

@@ -54,7 +54,7 @@ public NdArraySequence<U> elements(int dimensionIdx) {
 
   @Override
   public U withShape(Shape shape) {
-    if (shape == this.shape()) {
+    if (shape.equals(this.shape())) {
       return (U)this;
     }
     if (shape == null || shape.isUnknown() || shape.size() != this.shape().size()) {

ndarray/src/test/java/org/tensorflow/ndarray/SparseNdArrayTest.java

@@ -23,7 +23,10 @@
 import org.tensorflow.ndarray.impl.sparse.LongSparseNdArray;
 import org.tensorflow.ndarray.impl.sparse.ShortSparseNdArray;
 
-import static org.junit.jupiter.api.Assertions.*;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertThrows;
+import static org.junit.jupiter.api.Assertions.assertTrue;
 
 public class SparseNdArrayTest {
   long[][] indicesArray = {{0, 0}, {1, 2}, {2, 3}};