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/NdArrayTestBase.java

@@ -400,7 +400,7 @@ public void withShape() {
     assertEquals(originalShape, originalArray.shape());
     assertEquals(valueOf(10L), newArray.getObject(0, 0));
 
-    NdArray<T> sameArray = originalArray.withShape(originalShape);
+    NdArray<T> sameArray = originalArray.withShape(Shape.scalar());
     assertSame(originalArray, sameArray);
 
     assertThrows(IllegalArgumentException.class, () -> originalArray.withShape(Shape.of(2)));

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}};